hdu2896 AC自动机

hdu2896：

题意：有一些串是病毒串（模式串），标号1～n，给出m个待匹配串，问每个模式串中有哪些病毒串，从小到大输出，并求共有多少待匹配串含有病毒串

就是将AC自动机的模板改了一下，用模式串结尾结点记录下这是哪个模式串的结尾，在遍历时再排序输出即可

#include<stdio.h>
#include<string.h>
#include<queue>
#include<algorithm>
using namespace std;
const int maxm=520*250;

char s[10020],word[210];
bool vis[maxm];
int nxt[maxm][128-30],cnt[maxm],fail[maxm],size;
int ans[520],c;

int newnode(){
    memset(nxt[size],0,sizeof(nxt[size]));
    fail[size]=cnt[size]=0;
    return size++;
}

void insert(char s[],int k){
    int i,p=0;
    for(i=0;s[i];i++){
        int &x=nxt[p][s[i]-30];
        p=x?x:x=newnode();
    }
    cnt[p]=k;
}

void makenxt(){
    int i;
    queue<int>q;
    q.push(0);
    while(!q.empty()){
        int u=q.front();
        q.pop();
        for(i=0;i<128-30;i++){
            int v=nxt[u][i];
            if(v==0)nxt[u][i]=nxt[fail[u]][i];
            else q.push(v);    
            if(u&&v){
                fail[v]=nxt[fail[u]][i];
            }
        }
    }
}

bool query(char s[]){
    int d=0;
    bool f=0;
    for(int i=0;s[i];i++){
        d=nxt[d][s[i]-30];
        int tmp=d;
        while(tmp!=0){
            if(cnt[tmp]&&!vis[tmp]){
                f=1;
                vis[tmp]=1;
                ans[++c]=cnt[tmp];
            }
            tmp=fail[tmp];
        }
    }
    return f;
}

int main(){
    int n,m,i;
    while(scanf("%d",&n)!=EOF){
        memset(cnt,0,sizeof(cnt));
        size=0,newnode();
        for(i=1;i<=n;i++){
            scanf("%s",word);
            insert(word,i);
        }
        makenxt();
        scanf("%d",&m);
        int res=0;
        for(i=1;i<=m;i++){
            memset(vis,0,sizeof(vis));
            c=0;
            scanf("%s",s);
            if(query(s)){
                res++;
                sort(ans+1,ans+c+1);
                printf("web %d:",i);
                for(int j=1;j<=c;j++){
                    printf(" %d",ans[j]);
                }
                printf("
");
            }
        }
        printf("total: %d
",res);
    }
    return 0;
}