HDU2222 Keywords Search AC自动机

做了这题，才知道，所谓的keyword可能出现好多次。

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
char a[51];
#define maxn 500001
char c[1000010];
int ch[maxn][26];
int fail[maxn],last[maxn];
int val[maxn];
int T,n,tot,root,nex,cur,ans;
queue<int>p;
int newnode()
{
	memset(ch[tot],0,sizeof(ch[tot]));
	val[tot]=0;
	return tot++;
}
void insert(char *x)
{
	int len=strlen(x);
	int now=root;
	for(int i=0;i<len;i++)
	{
		int &temp=ch[now][x[i]-'a'];
		if(!temp)
			temp=newnode();
		now=temp;
	}
	val[now]++;
}
void getfail()
{
	fail[root]=root;
	last[root]=root;
	int i;
	for(i=0;i<26;i++)
	{
		nex=ch[root][i];
		if(nex)
		{
			fail[nex]=last[nex]=root;
			p.push(nex);
		}		
	}
	while(!p.empty())
	{
		cur=p.front();
		p.pop();
		for(i=0;i<26;i++)
		{
			nex=ch[cur][i];
			if(!nex)
				ch[cur][i]=ch[fail[cur]][i];
			else
			{
				fail[nex]=ch[fail[cur]][i];
				last[nex]=val[fail[nex]]?fail[nex]:last[fail[nex]];
				p.push(nex);
			}
		}
	}

}
void query(char *x)
{
	int len=strlen(x),i;
	ans=0;
	cur=root;
	for(i=0;i<len;i++)
	{
		cur=ch[cur][x[i]-'a'];
		int temp=cur;
		while(temp!=root&&val[temp])
		{
			ans+=val[temp];//不是ans++;
			val[temp]=0;
			temp=last[temp];			
		}
	}
	printf("%d\n",ans);

}
int main()
{
	scanf("%d",&T);
	while(T--)
	{
		scanf("%d",&n);
		int i;
		tot=0;
		root=newnode();
		for(i=0;i<n;i++)
		{
			scanf("%s",a);
			insert(a);
		}
		getfail();
		scanf("%s",c);
		query(c);
	}
	return 0;
}