AC自动机 hdu 2222 Keywords Search

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2222

　　今天木有题意，也木有分析，AC自动机裸题，省赛前敲一敲（抄一抄）代码。

AC代码：

#include<bits/stdc++.h>

using namespace std;

struct Aho{
    struct state{
        int next[26];
        int fail,cnt;
    }StateTable[500005];
    int size;
    queue<int>q;
    void init(){
        while(!q.empty()) q.pop();
        for(int i=0;i<500005;i++){
            memset(StateTable[i].next,0,sizeof(StateTable[i].next));
            StateTable[i].fail=StateTable[i].cnt=0;
        }
        size=0;
    }
    void insert(char *s){
        int n=strlen(s);
        int now=0;
        for(int i=0;i<n;i++){
            char c=s[i];
            if(!StateTable[now].next[c-'a']){
                StateTable[now].next[c-'a']=++size;
            }
            now=StateTable[now].next[c-'a'];
        }
        StateTable[now].cnt++;
    }
    void build(){
        StateTable[0].fail=-1;
        q.push(0);
        while(!q.empty()){
            int u=q.front();
            q.pop();
            for(int i=0;i<26;i++){
                if(StateTable[u].next[i]){
                    if(u==0){
                        StateTable[StateTable[u].next[i]].fail=0;
                    }
                    else {
                        int v=StateTable[u].fail;
                        while(v!=-1){
                            if(StateTable[v].next[i]){
                                StateTable[StateTable[u].next[i]].fail=StateTable[v].next[i];
                                break;
                            }
                            v=StateTable[v].fail;
                        }
                        if(v==-1){
                            StateTable[StateTable[u].next[i]].fail=0;
                        }
                    }
                    q.push(StateTable[u].next[i]);
                }
            }
        }
    }

    int get(int u){
        int res=0;
        while(u){
            res+=StateTable[u].cnt;
            StateTable[u].cnt=0;
            u=StateTable[u].fail;
        }
        return res;
    }
    int macth(char *s){
        int n=strlen(s);
        int res=0,now=0;
        for(int i=0;i<n;i++){
            char c=s[i];
            if(StateTable[now].next[c-'a']){
                now=StateTable[now].next[c-'a'];
            }
            else {
                int p=StateTable[now].fail;
                while(p!=-1&&StateTable[p].next[c-'a']==0){
                    p=StateTable[p].fail;
                }
                if(p==-1){
                    now=0;
                }
                else {
                    now=StateTable[p].next[c-'a'];
                }
            }
            if(StateTable[now].cnt){
                res+=get(now);
            }
        }
      return res;
    }
}aho;
int t,n;
char s[1000006];
int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    cin>>t;
    while(t--){
        aho.init();
        cin>>n;
        for(int i=1;i<=n;i++){
            cin>>s;
            aho.insert(s);
        }
        aho.build();
        cin>>s;
        cout<<aho.macth(s)<<endl;
    }
    return 0;
}