HDU - 2896 病毒侵袭 （AC自动机模板题）

病毒侵袭

 HDU - 2896 

当太阳的光辉逐渐被月亮遮蔽，世界失去了光明，大地迎来最黑暗的时刻。。。。在这样的时刻，人们却异常兴奋——我们能在有生之年看到500年一遇的世界奇观，那是多么幸福的事儿啊~~ 
但网路上总有那么些网站，开始借着民众的好奇心，打着介绍日食的旗号，大肆传播病毒。小t不幸成为受害者之一。小t如此生气，他决定要把世界上所有带病毒的网站都找出来。当然，谁都知道这是不可能的。小t却执意要完成这不能的任务，他说：“子子孙孙无穷匮也！”（愚公后继有人了）。 
万事开头难，小t收集了好多病毒的特征码，又收集了一批诡异网站的源码，他想知道这些网站中哪些是有病毒的，又是带了怎样的病毒呢？顺便还想知道他到底收集了多少带病毒的网站。这时候他却不知道何从下手了。所以想请大家帮帮忙。小t又是个急性子哦，所以解决问题越快越好哦~~ 

Input

第一行，一个整数N（1<=N<=500），表示病毒特征码的个数。 
接下来N行，每行表示一个病毒特征码，特征码字符串长度在20—200之间。 
每个病毒都有一个编号，依此为1—N。 
不同编号的病毒特征码不会相同。 
在这之后一行，有一个整数M（1<=M<=1000），表示网站数。 
接下来M行，每行表示一个网站源码，源码字符串长度在7000—10000之间。 
每个网站都有一个编号，依此为1—M。 
以上字符串中字符都是ASCII码可见字符（不包括回车）。 

Output

依次按如下格式输出按网站编号从小到大输出，带病毒的网站编号和包含病毒编号，每行一个含毒网站信息。 
web 网站编号: 病毒编号 病毒编号 … 
冒号后有一个空格，病毒编号按从小到大排列，两个病毒编号之间用一个空格隔开，如果一个网站包含病毒，病毒数不会超过3个。 
最后一行输出统计信息，如下格式 
total: 带病毒网站数 
冒号后有一个空格。 

Sample Input

3
aaa
bbb
ccc
2
aaabbbccc
bbaacc

Sample Output

web 1: 1 2 3
total: 1

题意：题意很好理解，先给你几个字符串，然后给你几个大长串，问你有大长串中有几个先前给定的字符串，简单明了，AC自动机问题，直接套模板，在建树的每个分支的结尾处end[]，标记上id ，因为是要匹配很多条，所以在查找的时候，不要让end[] = 0,用一个visit 数组标记，就可以重复使用了。在每次找到时，就用id[]保存。题目很简单，但是提交的时候直接RE了，我还以为数组开小了，改大之后依旧RE，最后重读了一遍题目才发现，原题说的ASCII码的字符，不是单纯的小写字母，给自己的粗心跪了orz。

AC 代码 ：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<queue>
#define rep(i,a,n) for(int i=a;i<n;i++)
#define per(i,a,n) for(int i=n-1;i>=a;i--)
#define fori(x) for(int i=0;i<x;i++)
#define forj(x) for(int j=0;j<x;j++)
#define memset(x,y) memset(x,y,sizeof(x))
#define memcpy(x,y) memcpy(x,y,sizeof(x))
#define INF 0x3f3f3f3f
const int maxn=100005;
typedef long long ll;
using namespace std;
int id[3];
int cas;
int tt;
int visit[maxn];
struct Trie
{
    int next[maxn][128],fail[maxn],end[maxn];
    int L, root;
    int newnode()
    {
        for(int i = 0; i < 128; i++)
            next[L][i] = -1;
        end[L++] = 0;
        return L - 1;
    }
    void init()
    {
        L = 0;
        root = newnode();
        return;
    }
    void insert(char *s)
    {
        int now = root;
        for(; *s; s++)
        {
            if(next[now][(int)*s] == -1)
                next[now][(int)*s] = newnode();
            now = next[now][(int)*s];
        }
        end[now] = cas++;
        return;
    }
    void build()
    {
        queue<int>Q;
        fail[root]=root;
        for(int i=0;i<128;i++)
            if(next[root][i]==-1)next[root][i]=root;
        else {
            fail[next[root][i]]=root;
            Q.push(next[root][i]);
        }
        while(!Q.empty()){
            int now=Q.front();
            Q.pop();
            for(int i=0;i<128;i++)
                if(next[now][i]==-1)
                next[now][i]=next[fail[now]][i];
            else {
                fail[next[now][i]]=next[fail[now]][i];
                Q.push(next[now][i]);
            }
        }
      }
    int query(char buf[])
    {
        int len=strlen(buf);
        int now=root;
        int res=0;
        for(int i=0;i<len;i++)
        {
            now=next[now][(int)buf[i]];
            int temp=now;
            while(temp!=root)
            {
                if(end[temp] && !visit[temp])
                {
                  id[res++]=end[temp];
                  visit[temp]=1;
                }
                temp=fail[temp];
            }
            if(res==3)
              break;
        }
        return res;
    }
    void debug(){
        for(int i=0;i<L;i++)
        {
            printf("id = %3d,fail = %3d,end = %3d,chi = [",i,fail[i],end[i]);
            for(int j=0;j<26;j++)
                printf("%2d",next[i][j]);
            printf("]\n");
        }
    }

};
Trie ac;
int main()
{
  int t;
  int result=0;
  scanf("%d",&t);
  char s[maxn];
  ac.init();
  cas=1;
  while(t--)
  {
    scanf("%s",s);
    ac.insert(s);
  }
  ac.build();
  int m;
  scanf("%d",&m);
  rep(i,1,m+1)
  {
    memset(id,0);
    memset(visit,0);
    scanf("%s",s);
    int cnt=ac.query(s);
    if(!cnt)
      continue;
    printf("web %d:",i);
    result++;
    sort(id,id+cnt);
    rep(j,0,cnt)
      printf(" %d",id[j]);
    printf("\n");
  }
  printf("total: %d\n",result);
  return 0;
}