首页 > 代码库 > HDU 2896 病毒侵袭 Trie图
HDU 2896 病毒侵袭 Trie图
题目大意:给一些病毒字符串,问一些网址中有哪些病毒。
思路:AC自动机挺裸的题,但是听说Trie图还好写,时间还快,以后就不写AC自动机了,直接啥题都上Trie图吧。
注意:此题输出结尾要加回车,否则会PE!
CODE:
#include <queue> #include <cstdio> #include <cstring> #include <iostream> #include <algorithm> using namespace std; struct Trie{ Trie *son[130],*fail; int id; Trie() { id = 0; memset(son,NULL,sizeof(son)); } }*root = new Trie(); int cnt,asks; char s[10010]; bool v[1010]; int ill[1010]; inline void Insert(char *s,int p) { Trie *now = root; while(*s != '\0') { if(now->son[*s] == NULL) now->son[*s] = new Trie(); now = now->son[*s]; ++s; } now->id = p; } void MakeTrieGraph() { static queue<Trie *> q; while(!q.empty()) q.pop(); for(int i = 0; i < 128; ++i) if(root->son[i] != NULL) root->son[i]->fail = root,q.push(root->son[i]); while(!q.empty()) { Trie *now = q.front(); q.pop(); for(int i = 0; i < 128; ++i) if(now->son[i] != NULL) { Trie *temp = now->fail; while(temp != root && temp->son[i] == NULL) temp = temp->fail; if(temp->son[i] != NULL) temp = temp->son[i]; now->son[i]->fail = temp; q.push(now->son[i]); } else now->son[i] = now->fail->son[i] ? now->fail->son[i]:root; } } inline void Ask(char *s) { Trie *now = root; while(*s != '\0') { if(now->son[*s] != NULL) now = now->son[*s]; for(Trie *temp = now; temp != root; temp = temp->fail) if(temp->id && !v[temp->id]) v[temp->id] = true,ill[++ill[0]] = temp->id; ++s; } } int main() { cin >> cnt; for(int i = 1; i <= cnt; ++i) { scanf("%s",s); Insert(s,i); } MakeTrieGraph(); cin >> asks; int ans = 0; for(int i = 1; i <= asks; ++i) { memset(v,false,sizeof(v)); memset(ill,0,sizeof(ill)); scanf("%s",s); Ask(s); if(ill[0]) { ++ans; sort(ill + 1,ill + ill[0] + 1); printf("web %d:",i); for(int j = 1; j <= ill[0]; ++j) printf(" %d",ill[j]); puts(""); } } printf("total: %d\n",ans); return 0; }
HDU 2896 病毒侵袭 Trie图
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。