P3796 【模板】AC自动机

传送门

AC自动机的模板

简单的理解就是字典树上的KMP

注意数组不要开太大

不然每次memset耗时太多

有一个小优化

每次走 fail 边找匹配时只有一些会更新答案

那么就可以把没用的fail边压缩掉

设 g[x] 表示从 x 点一直走 fail 边，走到的第一个有结束标记的点

那么找匹配时就只有要 g 边

然后就是模板了

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;
const int N=5e4+;
int len,n;
char s[];
inline void read_s()//文本串比较大，可以用快读加速
{
    char ch=getchar(); len=;
    while(ch<'a'||ch>'z') ch=getchar();
    while(ch>='a'&&ch<='z')
    {
        s[++len]=ch;
        ch=getchar();
    }
}
int c[N][],pd[N],fail[N],g[N],cnt;
char a[][]; int l[];
inline void clr()//清空AC自动机
{
    memset(c,,sizeof(c));
    memset(pd,,sizeof(pd));
    memset(fail,,sizeof(fail));
    memset(g,,sizeof(g));
    cnt=;
}
inline void ins(int num)//插入单词到自动机（跟字典树没区别）
{
    int u=;
    for(int i=;i<l[num];i++)
    {
        int v=a[num][i]-'a'+;
        if(!c[u][v]) c[u][v]=++cnt;
        u=c[u][v];
    }
    pd[u]=num;
}
queue <int> q;
inline void pre()//预处理fail和g
{
    for(int i=;i<=;i++)
        if(c[][i]) q.push(c[][i]);
    while(!q.empty())
    {
        int u=q.front(); q.pop();
        for(int i=;i<=;i++)
        {
            int v=c[u][i];
            if(!v) c[u][i]=c[fail[u]][i];//失配了直接走失配边
            else
            {
                fail[v]=c[fail[u]][i];
                g[v]= pd[fail[v]] ? fail[v] : g[fail[v]];//处理g
                q.push(v);
            }
        }
    }
}
int mx,ans[];
inline void query()//AC自动机匹配(好像跟字典树也没什么区别)
{
    memset(ans,,sizeof(ans));
    int u=;
    for(int i=;i<=len;i++)
    {
        u=c[u][s[i]-'a'+];
        if(pd[u]) ans[pd[u]]++;
        for(int j=g[u];j;j=g[j])
            ans[pd[j]]++;
        //计算匹配
    }
    mx=;
    for(int i=;i<=n;i++)
        mx=max(mx,ans[i]);
    cout<<mx<<endl;
    for(int i=;i<=n;i++)
        if(ans[i]==mx)
            printf("%s\n",a[i]);
}
int main()
{
    while()
    {
        scanf("%d",&n); if(!n) break;
        clr();
        for(int i=;i<=n;i++)
        {
            scanf("%s",a[i]);
            l[i]=strlen(a[i]);
            ins(i);
        }
        pre();
        read_s();
        query();
    }
    return ;
}