hdoj2896（AC自动机简单题）

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<queue>
using namespace std;

const int maxn=1e5+5;
const int maxs=1e4+5;
int n,m,cnt,ans,trie[maxn][130],key1[maxn],key2[maxn],fail[maxn];
int res[5],cnt1;
char s[maxs];

void build(char *s,int k){
    int len=strlen(s),u=0;
    for(int i=0;i<len;++i){
        int t=s[i];
        if(!trie[u][t]){
            ++cnt;
            memset(trie[cnt],0,sizeof(trie[cnt]));
            key1[cnt]=0,fail[0]=0;
            trie[u][t]=cnt;
        }
        u=trie[u][t];
    }
    key1[u]=k;
}

void get_fail(){
    queue<int> que;
    for(int i=0;i<130;++i){
        if(trie[0][i]){
            fail[trie[0][i]]=0;
            que.push(trie[0][i]);
        }
    }
    while(!que.empty()){
        int u=que.front();que.pop();
        for(int i=0;i<130;++i){
            if(trie[u][i]){
                fail[trie[u][i]]=trie[fail[u]][i];
                que.push(trie[u][i]);
            }
            else{
                trie[u][i]=trie[fail[u]][i];
            }
        }
    }
}

void query(char *s){
    int len=strlen(s),u=0;
    for(int i=0;i<len;++i){
        int t=s[i];
        u=trie[u][t];
        for(int j=u;j&&key2[j]!=-1;j=fail[j]){
            if(key2[j]) res[++cnt1]=key2[j];
            key2[j]=-1;
        }
    }
}

int main(){
    scanf("%d",&n);
    getchar();
    cnt=0,ans=0;
    memset(trie[0],0,sizeof(trie[0]));
    key1[0]=0;
    for(int i=1;i<=n;++i){
        scanf("%[^\n]",s);
        getchar();
        build(s,i);
    }
    fail[0]=0;
    get_fail();
    scanf("%d",&m);
    getchar();
    for(int i=1;i<=m;++i){
        for(int j=0;j<=cnt;++j)
            key2[j]=key1[j];
        scanf("%[^\n]",s);
        getchar();
        cnt1=0;
        query(s);
        if(cnt1){
            ++ans;
            sort(res+1,res+cnt1+1);
            printf("web %d:",i);
            for(int j=1;j<=cnt1;++j)
                printf(" %d",res[j]);
            printf("\n");
        }
    }
    printf("total: %d\n",ans);
    return 0;
}