AC自动机模板 hdu2222

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <string.h>
#include <queue>                //AC自动机模板题（动态建立） hdu2222
using namespace std;
char a[1000005], b[55];
struct trie
{
trie *next[26];
trie *fail;    //失败指针,类似于kmp中的next
    int g;
trie()
{
for(int t=0; t<26; ++t)
next[t]=NULL;
fail=NULL;
g=0;
}
}*root;
void create(char *p)
{
int t=0, k, j=strlen(p);
trie *q=root;
while(t<j)
{
k=p[t]-'a';
if(!q->next[k])
{
q->next[k]=new trie();
}
q=q->next[k];
t++;
}
q->g++;    //若重复也算，则加上去
    return ;
}
void makefail()
{
queue<trie *> q;
q.push(root);
while(!q.empty())
{
trie *p=q.front(), *temp;
q.pop();
for(int t=0; t<26; ++t)
{
if(p->next[t])
{
temp=p;
if(p==root)
{
temp->next[t]->fail=p;
}
else
{
while(temp->fail)
{
if(temp->fail->next[t])
{
p->next[t]->fail=temp->fail->next[t];
break;
}
temp=temp->fail;
}
if(!temp->fail)
{
p->next[t]->fail=root;
}
}
q.push(p->next[t]);
}
}
}
return ;
}
int f(char  *p)
{
int t, k, s=0, j=strlen(p);
trie *q=root;
for(t=0; t<j; ++t)
{
k=p[t]-'a';
while(q!=root&&q->next[k]==NULL)
{
q=q->fail;
}
q=q->next[k];
if(!q)q=root;
trie *temp=q;
while(temp!=root&&temp->g)
{
s+=temp->g;   //加上重复的
temp->g=0;
temp=temp->fail;
}
}
return s;
}
int main()
{
int n, t, T;
scanf("%d", &T);
while(T--)
{
root=new trie();
scanf("%d", &n);
for(t=0; t<n; ++t)
{
scanf("%s", b);
create(b);
}
makefail();
scanf("%s", a);
printf("%d\n", f(a));
}
return 0;
}