【HDU】2222 Keywords Search（AC自动机）

题目

传送门：QWQ

---

分析

$ AC $自动机模板，黈力的码风真的棒极了，这是我抄他的。

还有 题号不错

---

代码

#include <cstdio>
#include <cstring>
#define N 500007
using namespace std;
int n,rt,newp,i,m,l,r,son[N][],fail[N],cur,j,end[N],q[N],vis[N],ans[N],end2[N],end3[N];
char s[];
int find(int cur,int i){
    if(!cur)return rt;
    if(son[cur][i])return son[cur][i];
    return fail[son[cur][i]]=find(fail[cur],i);
}
int main(){
    int T; scanf("%d",&T);
    while(T--)
    {
        memset(son,,sizeof(son));memset(fail,,sizeof(fail));memset(vis,,sizeof(vis));
        memset(q,,sizeof(q));memset(end,,sizeof(end));memset(ans,,sizeof(ans));
        memset(end2,,sizeof(end2)); memset(end3,,sizeof(end3));
    scanf("%d",&n);
    rt=newp=;
    for(i=;i<=n;i++){
        scanf("%s",s+);
        m=strlen(s+);
        for(cur=rt,j=;j<=m;j++){
            if(!son[cur][s[j]-'a'])
                son[cur][s[j]-'a']=++newp;
            cur=son[cur][s[j]-'a'];
        }
        end[cur]=i;
        end2[cur]++;
        end3[i]=cur;
    }
    for(q[l=r=]=rt;l<=r;l++){
        for(i=;i<;i++)
        if(son[q[l]][i]){
            fail[son[q[l]][i]]=find(fail[q[l]],i);
            q[++r]=son[q[l]][i];
        }
    }
    scanf("%s",s+);
    m=strlen(s+);int ansx=;
    for(i=,cur=rt;i<=m;i++)
        cur=find(cur,s[i]-'a'),vis[cur]++;
    for(i=r;i>=;i--){
        vis[fail[q[i]]]+=vis[q[i]];
        if(end[q[i]])
        {
            if(vis[q[i]]!=) ansx+=end2[q[i]];
        }
    }
    printf("%d\n",ansx);
    }
    return ;
}
/*
1
10
ab aba bba a aa a a aba b b
ababbaba
*/