HDU 3065 病毒侵袭持续中 (AC自动机)

题意:找出模式串出现的种类是什么以及它们的次数

思路:拿一个数组存end的出现次数。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<algorithm>
#include<queue>
#define maxn 50005
char s[2000005],a[1005][105];
int n,num[1005];
struct Trie{
    int fail[maxn],end[maxn],next[maxn][130],L,root;
    int newnode(){
        for (int i=0;i<128;i++)
         next[L][i]=-1;
        end[L++]=-1;
        return L-1; 
    }
    void clear(){
        L=0;
        root=newnode();
    }
    void insert(char s[],int id){
        int now=root,len=strlen(s);
        for (int i=0;i<len;i++){
            if (next[now][s[i]]==-1) next[now][s[i]]=newnode();
            now=next[now][s[i]];
        }
        end[now]=id;
    }
    void build(){
        std::queue<int>Q;
        int now=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()){
            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]);
                } 
            }
        }
    }
    void query(char s[]){
        int len=strlen(s),now=root;
        for (int i=0;i<=n;i++)
         num[i]=0;
        for (int i=0;i<len;i++){
            now=next[now][s[i]];
            int tmp=now;
            while (tmp!=root){
                if (end[tmp]!=-1)
                num[end[tmp]]++;
                tmp=fail[tmp];
            }
        }
        for (int i=1;i<=n;i++)
         if (num[i]){
             int len=strlen(a[i]);
             for (int j=0;j<len;j++)
              printf("%c",a[i][j]);
             printf(": %d",num[i]);
            printf("
");  
         }
    }
}ac;
int main(){
    while (scanf("%d",&n)!=EOF){
        ac.clear();
        for (int i=1;i<=n;i++){
            scanf("%s",a[i]);
            ac.insert(a[i],i);
        }
        ac.build();
        scanf("%s",s);
        ac.query(s);
    }
}