hdu 3065 病毒侵袭持续中(AC自动机)

Description:

小t非常感谢大家帮忙解决了他的上一个问题。然而病毒侵袭持续中。在小t的不懈努力下，他发现了网路中的“万恶之源”。这是一个庞大的病毒网站，他有着好多好多的病毒，但是这个网站包含的病毒很奇怪，这些病毒的特征码很短，而且只包含“英文大写字符”。当然小t好想好想为民除害，但是小t从来不打没有准备的战争。知己知彼，百战不殆，小t首先要做的是知道这个病毒网站特征：包含多少不同的病毒，每种病毒出现了多少次。大家能再帮帮他吗？

Input:

第一行，一个整数N（1<=N<=1000），表示病毒特征码的个数。 
接下来N行，每行表示一个病毒特征码，特征码字符串长度在1―50之间，并且只包含“英文大写字符”。任意两个病毒特征码，不会完全相同。 
在这之后一行，表示“万恶之源”网站源码，源码字符串长度在2000000之内。字符串中字符都是ASCII码可见字符（不包括回车）。 

Output

按以下格式每行一个，输出每个病毒出现次数。未出现的病毒不需要输出。 
病毒特征码: 出现次数 
冒号后有一个空格，按病毒特征码的输入顺序进行输出。 

3

AA

BB

CC

ooxxCC%dAAAoen....END

Output

AA: 2

CC: 1

代码：

#include<cstdio>#include<cstring>#include<iostream>#include<queue>#include<cstdlib>using namespace std;#define maxn (26)int n;int ans[1000+10];char a[1000+1][50+1];char s[2000000+10];struct trie{int col,num;trie *next[maxn],*fail;};queue<trie *>q;trie *get_trie(){trie *temp = new trie;temp->col = 0;temp->num = 0;for(int i=0;i<maxn;i++)temp->next[i] = NULL;return temp;}void creat(trie *root,char *s,int cnt){trie *temp = root;int l = strlen(s);for(int i = 0;i < l; i++){int pos=s[i]-'A';if(temp->next[pos]==NULL)temp->next[pos] = get_trie();temp=temp->next[pos];}temp->col++;temp->num = cnt;}void built_ac(trie *root){root->fail = NULL;q.push(root);while(!q.empty()){trie *p = q.front();q.pop();trie *temp = NULL;for(int i = 0;i < maxn; i++){if(p->next[i] != NULL){if(p == root)p->next[i]->fail = root;else{temp = p->fail;while(temp != NULL){if(temp->next[i] != NULL){p->next[i]->fail = temp->next[i];break;}temp = temp->fail;}if(temp == NULL)p->next[i]->fail = root;}q.push(p->next[i]);}}}}void find(trie *root,char *s){int l = strlen(s);trie *p = root;for(int i = 0;i < l; i++){if(s[i]>='A'&&s[i]<='Z'){int pos = s[i]-'A';while(p->next[pos]==NULL && p != root)p=p->fail;p = p->next[pos];if(p == NULL)p=root;trie *temp = p;while(temp != root && temp->col > 0){ans[temp->num]++;temp = temp->fail;}}else p = root;}for(int i = 1;i <= n; i++){if(ans[i] > 0){printf("%s: %d\n",a[i],ans[i]);}}}void read(){trie *root;root = get_trie();while(scanf("%d",&n)==1){memset(ans,0,sizeof(ans));for(int i = 1;i <= n; ++i){scanf("%s",a[i]);creat(root,a[i],i);}built_ac(root);scanf("%s",s);find(root,s);    }}int main(){read();return 0;}