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;
}