【HDU2896】病毒侵袭-AC自动机模板题

测试地址：病毒侵袭

做法：就是AC自动机的模板题，网上教程很多，这里不再赘述。但是我写完之后交上去发现格式错误，结果在文末多输出个空行就对了...好迷啊...

以下是本人代码：

#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <queue>
using namespace std;
char s[1000010];
int n,m,tot,root,now,total=0;
bool vis[510];

struct
{
  int next[100010][128],fail[100010],end[100010];
  
  int newnode()
  {
    end[++tot]=0;fail[tot]=0;
	for(int i=0;i<128;i++)
	  next[tot][i]=0;
    return tot;
  }
  
  void destroy()
  {
    tot=0;
	root=newnode();
  }
  
  void insert()
  {
    int v=root,i=0;
	while(s[i]!=0)
	{
	  int f=v;
	  v=next[v][(int)s[i]];
	  if (v==0) v=newnode(),next[f][(int)s[i]]=v;
	  i++;
	}
	end[v]=now;
  }
  
  void build()
  {
    queue<int> q;
	q.push(root);
	fail[root]=root;
	while(!q.empty())
	{
	  int v=q.front();q.pop();
	  for(int i=0;i<128;i++)
	    if (next[v][i]!=0)
		{
		  if (v==root) fail[next[v][i]]=root;
		  else
		  {
		    int to=fail[v];
			while(to!=root)
			{
			  if (next[to][i]==0) to=fail[to];
			  else break;
			}
			if (next[to][i]!=0) fail[next[v][i]]=next[to][i];
			else fail[next[v][i]]=root;
		  }
		  q.push(next[v][i]);
		}
	}
  }
  
  void query()
  {
    int v=root,i=0;
	bool flag=0;
    memset(vis,0,sizeof(vis));
	while(s[i]!=0)
	{
	  while(next[v][(int)s[i]]==0&&v!=root) v=fail[v];
	  if (next[v][(int)s[i]]!=0)
	  {
	    v=next[v][(int)s[i]];
		int to=v;
	    while(end[to]!=0)
		{
		  flag=1;
		  vis[end[to]]=1;
		  to=fail[to];
		}
	  }
	  i++;
	}
	if (flag)
	{
	  printf("web %d:",now);
	  for(int i=1;i<=n;i++)
	    if (vis[i]) printf(" %d",i);
	  printf("\n");
	  total++;
	}
  }
}AC;

int main()
{
  scanf("%d",&n);getchar();
  AC.destroy();
  for(now=1;now<=n;now++)
  {
    scanf("%s",s);
	AC.insert();
  }
  scanf("%d",&m);getchar();
  AC.build();
  for(now=1;now<=m;now++)
  {
    scanf("%s",s);
	AC.query();
  }
  printf("total: %d\n",total);
  
  return 0;
}