【模板】AC自动机（洛谷3808&洛谷3796）

清明时节雨纷纷，省选只能当路人。
最近在狂补算法，AC自动机还是比较难理解。找到了洛谷日报上比较容易理解的版本，才勉强理解。
https://www.luogu.org/blog/42196/qiang-shi-tu-xie-ac-zi-dong-ji
千万记得前置技能trie树和KMP！
我博客里也有有关KMP的感想。（顺带安利一发
https://www.cnblogs.com/NightRaven/p/10554401.html
至于trie树，是个比较简单（应该）的数据结构，稍微看看就会了8（大概）（高端操作依然不会）
————————————————
总之以下是洛谷上简单版的AC自动机（加强版请再往下翻翻

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define pb push_back
using namespace std;
const int maxn=1000005;
queue<int> q;
struct ac_automation
{
    int tr[maxn][30],cnt,fail[maxn],val[maxn];
    void clear_ac()
    {
        memset(fail,0,sizeof(fail));
        memset(val,0,sizeof(val));
        cnt=0;
    }
    void add(char *s)
    {
        int len=strlen(s),k=0;
        rep(i,0,len-1)
        {
            int c=s[i]-'a';
            if(!tr[k][c]) tr[k][c]=++cnt;
            k=tr[k][c];
        }
        val[k]++;
    }
    void built()
    {
        rep(i,0,25) if(tr[0][i]) q.push(tr[0][i]);
        while(!q.empty())
        {
            int x=q.front();
            q.pop();
            rep(i,0,25) if(tr[x][i])
            fail[tr[x][i]]=tr[fail[x]][i],q.push(tr[x][i]);
            else tr[x][i]=tr[fail[x]][i];
        }
    }
    int query(char *s)
    {
        int len=strlen(s),k=0,ans=0;
        rep(i,0,len-1)
        {
            int c=s[i]-'a';
            k=tr[k][c];
            for(int j=k;j && val[j]!=-1;j=fail[j]) ans+=val[j],val[j]=-1;//记得看题，每个词只记一次，若多次就不用赋值为-1.
        }
        return ans;
    }
} acauto;
int main()
{
    int n;
    scanf("%d",&n);
    acauto.clear_ac();
    rep(i,1,n)
    {
        char c[maxn];
        scanf("%s",c);
        acauto.add(c);
    }
    char p[maxn];
    scanf("%s",p);
    acauto.built();
    printf("%d",acauto.query(p));
    return 0;
}

下面是加强版

#include<bits/stdc++.h>
#define rep(i,k,n) for(int i=k;i<=n;i++)
#define pb push_back
using namespace std;
const int maxn=1000005;
queue<int> q; 
char c[155][75];
int ans[155];
struct ac_automation
{
    int tr[maxn][30],cnt,fail[maxn],val[maxn];
    void clear_ac()
    {
        memset(fail,0,sizeof(fail));
        memset(val,0,sizeof(val));
        memset(tr,0,sizeof(tr));
        cnt=0;
    }
    void add(char *s,int j)
    {
        int len=strlen(s),k=0;
        rep(i,0,len-1)
        {
            int c=s[i]-'a';
            if(!tr[k][c]) tr[k][c]=++cnt;
            k=tr[k][c];
        }
        val[k]=j;
    }
    void built()
    {
        rep(i,0,25) if(tr[0][i]) q.push(tr[0][i]),fail[tr[0][i]]=0;
        while(!q.empty())
        {
            int x=q.front();
            q.pop();
            rep(i,0,25) if(tr[x][i])
            fail[tr[x][i]]=tr[fail[x]][i],q.push(tr[x][i]);
            else tr[x][i]=tr[fail[x]][i];
        }
    }
    void query(char *s)
    {
        int len=strlen(s),k=0;
        rep(i,0,len-1)
        {
            int c=s[i]-'a';
            k=tr[k][c];
            for(int j=k;j;j=fail[j]) ans[val[j]]++;
        }
    }
} acauto;
int main()
{
    int n;
    scanf("%d",&n);
    while(n)
    {
        memset(ans,0,sizeof(ans)); 
        acauto.clear_ac();
        rep(i,1,n)
        {
            scanf("%s",c[i]);
            acauto.add(c[i],i);
        }
        char p[maxn];
        scanf("%s",p);
        acauto.built();acauto.query(p);
        int maxo=-1;
        rep(i,1,n) maxo=max(maxo,ans[i]);
        printf("%d\n",maxo);
        rep(i,1,n) if(ans[i]==maxo) printf("%s\n",c[i]); 
        scanf("%d",&n);
    }
    return 0;
}

最近严重意识到自己菜的真实，很多算法都不会，又有很多的忘记了，DP也菜的不行。。。最近加油吧。

转载于:https://www.cnblogs.com/NightRaven/p/10561412.html