【洛谷3796】【模板】AC自动机（加强版）

最新推荐文章于 2022-11-09 05:15:59 发布

chenxiaoran666

最新推荐文章于 2022-11-09 05:15:59 发布

阅读量312

点赞数

CC 4.0 BY-SA版权

分类专栏：洛谷 AC自动机文章标签： AC自动机

本文链接：https://blog.youkuaiyun.com/chenxiaoran666/article/details/81876003

洛谷同时被 2 个专栏收录

46 篇文章

订阅专栏

AC自动机

6 篇文章

订阅专栏

本文介绍了一道AC自动机的模板题，通过构建AC自动机来找出文本串中出现次数最多的模式串。首先将模式串插入Trie树，并记录每个模式串对应的节点。接着构建AC自动机的失配指针，最后遍历文本串统计每个模式串的出现次数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

点此看题面

大致题意： 一道模板题，给你 $N$ 个模式串和一个文本串，要你求出在文本串中出现次数最多的若干个模式串并输出它们。

$A C$ 自动机

都说了是 $A C$ 自动机的模板题，做法肯定是 $A C$ 自动机。

$L i n k$

$A C$ 自动机 详见博客初学AC自动机

题解

我们可以考虑在将每个模式串插入 $T r i e$ 后，记录下每个模式串最后到达的节点。

然后，在 $A C$ 自动机时，将每一个经过的节点的访问次数加1。

最后统计答案时，只要求出最后到达的节点被访问次数最大的若干个模式串并输出即可，应该可以算是一道比较裸的模板题。

代码

#include<bits/stdc++.h>
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define LL long long
#define swap(x,y) (x^=y,y^=x,x^=y)
#define pc(ch) (pp_<100000?pp[pp_++]=(ch):(fwrite(pp,1,100000,stdout),pp[(pp_=0)++]=(ch)))
#define N 150
#define SUM 1000000
int pp_=0;char pp[100000];
using namespace std;
int n,tot=1,rt=1,ans=0,f[N+5];
struct Trie
{
    int Son[26],Vis,Next;
}node[SUM+5];
string s[N+5],st;
queue<int> q;
vector<int> res;
inline void write(int x)
{
    if(x>9) write(x/10);
    pc(x%10+'0');
}
inline void ps(string x)
{
    register int i,len=x.length();
    for(i=0;i<len;++i) pc(x[i]);
}
inline void Insert(int pos,string s)//将编号为pos的模式串s插入字典树中
{
    register int i;int x=rt;
    for(i=0;i<s.length();++i)
    {
        int p=s[i]-97;
        if(!node[x].Son[p]) node[x].Son[p]=++tot;
        x=node[x].Son[p];
    }
    f[pos]=x;//记录当前模式串最后到达的节点
}
inline void GetNext()//求出失配指针
{
    register int i,k;q.push(rt);
    while(!q.empty())
    {
        k=q.front(),q.pop();
        for(i=0;i<26;++i)
        {
            if(k^rt)
            {
                if(!node[k].Son[i]) node[k].Son[i]=node[node[k].Next].Son[i];
                else node[node[k].Son[i]].Next=node[node[k].Next].Son[i],q.push(node[k].Son[i]);
            }
            else 
            {
                if(!node[k].Son[i]) node[k].Son[i]=rt;
                else node[node[k].Son[i]].Next=rt,q.push(node[k].Son[i]);
            }
        }
    }
}
inline void AC_Automation()//AC自动机的核心代码
{
    register int i;int x=rt,len=st.length();
    for(i=0;i<len;++i)
    {
        if(!(x=node[x].Son[st[i]-97])) {x=rt;continue;}
        int p=x;
        while(p^rt) ++node[p].Vis,p=node[p].Next;//将每一个节点的访问次数加1
    }
}
int main()
{
    register int i,j;
    while(~scanf("%d",&n)&&n)
    {
        for(ans=0,i=1;i<=tot;++i) for(node[i].Next=node[i].Vis=j=0;j<26;++j) node[i].Son[j]=0;
        for(tot=i=1;i<=n;++i) cin>>s[i],Insert(i,s[i]);
        for(cin>>st,GetNext(),AC_Automation(),i=1;i<=n;++i) 
        {
            if(node[f[i]].Vis>ans) ans=node[f[i]].Vis,res.clear(),res.push_back(i);//比较当前模式串所对应的Trie上的节点被访问的次数与ans的大小，如果当前模式串出现的次数大于ans，就更新ans
            else if(node[f[i]].Vis==ans) res.push_back(i);//否则，如果当前模式串出现的次数等于ans，就将当前模式串加入最后要输出的数组中
        }
        for(write(ans),pc('\n'),i=0;i<res.size();++i) ps(s[res[i]]),pc('\n');
    }
    return fwrite(pp,1,pp_,stdout),0;
}