HDU2896 病毒侵袭 AC自动机

注意输入时要用gets();

#include<stdio.h>
#include<string.h>
#include<queue>
using namespace std;
int n,m;
#define maxn 125000
#define tt 128
int ch[maxn][tt],fail[maxn],last[maxn],val[maxn],tot,root,save[10],bian[10],ans,ge;
char s[10010];
int newnode()
{
	memset(ch[tot],-1,sizeof(ch[tot]));
	val[tot]=0;
	return tot++;
}
void insert(char *x,int v)
{
	int cur=root,len=strlen(x);
	for(int i=0;i<len;i++)
	{
		if(ch[cur][x[i]]==-1)
			ch[cur][x[i]]=newnode();
		cur=ch[cur][x[i]];
	}
	val[cur]=v;
}
bool query(char *x)
{
	int cur=root,len=strlen(x);
	ans=0;
	for(int i=0;i<len;i++)
	{
		cur=ch[cur][x[i]];
		int temp=cur;
		while(temp!=root&&val[temp]&&ans<3)
		{
			save[ans]=val[temp];
			val[temp]=0;			
			bian[ans++]=temp;
			temp=last[temp];
		}
	}
	if(ans>0)return true;
	else return false;
}
void getfail()
{
	queue<int>q;
	int cur;
	cur=root;
	fail[root]=last[root]=root;
	for(int i=0;i<tt;i++)
	{
		if(ch[root][i]!=-1)
		{
			fail[ch[root][i]]=last[ch[root][i]]=0;
			q.push(ch[root][i]);
		}		
		else ch[root][i]=0;
	}
	while(!q.empty())
	{
		cur=q.front();
		q.pop();
		for(int i=0;i<tt;i++)
		{
			if(ch[cur][i]==-1)
				ch[cur][i]=ch[fail[cur]][i];
			else
			{
				fail[ch[cur][i]]=ch[fail[cur]][i];
				last[ch[cur][i]]=val[fail[cur]]?fail[ch[cur][i]]:last[fail[ch[cur][i]]];
				q.push(ch[cur][i]);
			}
		}
	}
}
void ex(int i,int j)
{
	if(save[i]>save[j])
	{
		int x=save[j];
		save[j]=save[i];
		save[i]=x;
		x=bian[j];
		bian[j]=bian[i];
		bian[i]=x;
	}
}
int main()
{
	while(~scanf("%d",&n))
	{
		int i,k,j;
		tot=0;
		root=newnode();
		getchar();
		for(i=1;i<=n;i++)
		{
			gets(s);
			insert(s,i);
		}
		getfail();
		scanf("%d",&m);
		ge=0;
		for(i=1;i<=m;i++)
		{
			scanf("%s",&s);
			if(query(s))
			{
				ge++;
				printf("web %d:",i);
				for(k=0;k<ans;k++)
					for(j=k+1;j<ans;j++)
						ex(k,j);
				for(k=0;k<ans;k++)
				{
					printf(" %d",save[k]);
					val[bian[k]]=save[k];
				}
				printf("\n");
			}
		}
		printf("total: %d\n",ge);
	}
	return 0;
}