HDU 3065

AC自动机模板题

#include <iostream>
#include <cstdio>
#include <cstring>
#include <string.h>
#include <string>
#include <queue>
using namespace std;
const int N=1002;
char str[2000005],s1[N][52];
int ch[50010][26];
int val[50010],sz,n,root,fail[50010],count1[N];
int newnode(){
    memset(ch[sz],-1,sizeof(ch[sz]));
    val[sz]=-1;
    return sz++;
}
void init(){
    sz=0;
    root=newnode();
}
void insert(char* st,int id){
    int len=strlen(st);
    int now=root;
    for(int i=0;i<len;i++){
        int &u=ch[now][st[i]-'A'];
        if(u==-1)u=newnode();
        now=u;
    }
    val[now]=id;
    count1[id]=0;
}
void getfail(){
    queue<int> q;
    fail[root] = root;
    for(int i=0;i<26;i++){
        int &u=ch[root][i];
        if(u!=-1){
            fail[u]= 0;
            q.push(u);
        }
        else u=root;
    }
    while(!q.empty()){
        int now=q.front();
        q.pop();
        for(int i=0;i<26;i++){
            int u=ch[now][i];
            if(u==-1)ch[now][i]=ch[fail[now]][i];
            else {
                fail[u]=ch[fail[now]][i];
                q.push(u);
            }
        }
    }
}
void  query(char *s){
    int len=strlen(s);
    int now=root;
    for(int i=0;i<len;i++){
        if(s[i]>'Z'||s[i]<'A')now=root;
        else {
            now=ch[now][s[i]-'A'];
            int tem=now;
            while(tem!=root){
                if(val[tem]!=-1)count1[val[tem]]++;
                tem=fail[tem];
            }
        }
    }
}
int main(){
    int i;
    while(scanf("%d",&n)!=EOF){
        init();
        for(i=1;i<=n;i++){
            scanf("%s",s1[i]);
            insert(s1[i],i);
        }
        scanf("%s",str);
        getfail();
        query(str);
        for(i=1;i<=n;i++){
            if(count1[i])
            printf("%s: %d
",s1[i],count1[i]);
        }
    }
    return 0;
}