hdu 2896 病毒侵袭 ac自动机模板题

和上一道题不同的是这个题目的mark数组是标记一下到底是第几个串，因为匹配串中不同模式串个数不大于3，不过有可能相同串个数很多- -虽然我数组只开了10过了...还有就是字符的个数从26个飙升到127...所以建立字典树的时候需要做一些小的改动。查找到某个模式串标记一下，排序去重输出即可...

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<cstdlib>
#include<algorithm>
#include<map>
#include<queue>
using namespace std;
int ans[10];
#define N 1000000
struct Trie
{
    int next[N][130],fail[N],mark[N];
    int root,L;
    int newnode()
    {
        memset(next[L],-1,sizeof(next[L]));
        mark[L]=0;
        fail[L++]=-1;
        return L-1;
    }
    void go()
    {
        L=0;
        root=newnode();
    }
    void init(char *v,int id)
    {
        int p=root;
        for(int i=0;v[i];i++)
        {
            int tep=v[i];
            if(next[p][tep]==-1) next[p][tep]=newnode();
            p=next[p][tep];
        }
        mark[p]=id;
    }
    void bfs()
    {
        queue<int>q;
        q.push(root);
        while(!q.empty())
        {
            int p=q.front();
            q.pop();
            for(int i=0;i<127;i++)
            {
                if(next[p][i]==-1)
                {
                    if(p==root) next[p][i]=root;
                    else next[p][i]=next[fail[p]][i];
                }
                else
                {
                    if(p==root) fail[next[p][i]]=root;
                    else fail[next[p][i]]=next[fail[p]][i];
                    q.push(next[p][i]);
                }
            }
        }
    }
    int finde(char *v)
    {
        int k=0;
        int p=root,q;
        for(int i=0;v[i];i++)
        {
            int tep=v[i];
            while(next[p][tep]==-1&&p!=root)
                p=fail[p];
            p=next[p][tep];
            if(p==-1) p=root;
            q=p;
            while(q!=root)
            {
                if(mark[q]!=0)
                ans[k++]=mark[q];
                q=fail[q];
            }
        }
        return k;
    }
}ac;
int main()
{
    int n,m;
    char v[30000];
    while(scanf("%d",&n)!=EOF)
    {
        ac.go();
        for(int i=1;i<=n;i++)
        {
            scanf("%s",v);
            ac.init(v,i);
        }
        ac.bfs();
        int cnt=0;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            memset(ans,0,sizeof(ans));
            scanf("%s",v);
            int k=ac.finde(v);
            if(k)
            {
                printf("web %d:",i);
                sort(ans,ans+k);
                for(int j=0; j<k; j++)
                {
                    if(ans[j]!=ans[j+1])  printf(" %d",ans[j]);  //为了去重
                }
                puts("");
                cnt++;
            }
        }
        printf("total: %d\n",cnt);
    }
    return 0;
}