AC自动机

AC自动机

注意：Root不要设为0号点（我也不知道为什么，反正过不了样例，大概是我的写法不能将0号点作为根）

#include<cstdio>
#include<cstring>
#include<iostream>
#include<queue>
using namespace std;
char Mode[1000010];int tot,Root=0;
struct node{int son[26],num,Fail;}trie[500100];
void Ins(char s[55]){
    int p=Root,len=strlen(s);
    for(int i=0;i<len;i++){
        int t=s[i]-'a';
        if(!trie[p].son[t])trie[p].son[t]=++tot;
        p=trie[p].son[t];
    }
    trie[p].num=1;
}
void Build(){
    queue<int>q;q.push(Root);
    while(!q.empty()){
        int tmp=q.front();q.pop();
        for(int i=0;i<26;i++){
            if(!trie[tmp].son[i])continue;
            int Next=trie[tmp].son[i],p;
            for(p=trie[tmp].Fail;p;p=trie[p].Fail)
                if(trie[p].son[i])
                    {trie[Next].Fail=trie[p].son[i];break;}
            if(!p)trie[Next].Fail=Root;
            q.push(Next);
        }
    }
}
int Query(){
    int p=Root,cnt=0,len=strlen(Mode);
    for(int i=0;i<len;i++){
        int t=Mode[i]-'a';
        while(!trie[p].son[t] && p)p=trie[p].Fail;
        p=trie[p].son[t];
        if(!p)p=Root;
        for(int tmp=p;tmp!=Root && trie[tmp].num!=-1;tmp=trie[tmp].Fail)
            {cnt+=trie[tmp].num,trie[tmp].num=-1;}
    }
    return cnt;
}
int main(){
    int n;scanf("%d",&n);
    while(n--){char t[55];scanf("%s",t);Ins(t);}
    Build();
    scanf("%s",Mode);
    printf("%d",Query());
    return 0;

}