P3796 【模板】AC自动机（加强版）

最新推荐文章于 2022-06-17 13:43:24 发布

weixin_30780649

最新推荐文章于 2022-06-17 13:43:24 发布

阅读量51

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/lovewhy/p/9633565.html

本文详细介绍了AC自动机的实现原理及其在字符串匹配问题中的应用。通过具体的代码示例，展示了如何构建AC自动机并进行模式串的插入、失败指针的建立以及查询操作。适用于对AC自动机原理及实现感兴趣的读者。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<queue>
using namespace std;

const int N=1e6+5;
const int M=2e4;

int n;
char t[N],p[155][75];

int nxt[M][26];
int id[M],fail[M],last[M];
int now_node;
struct ANS
{
    int id,cnt;
    bool operator < (const ANS &A) const
    {
        return cnt==A.cnt?id<A.id:cnt>A.cnt;
    }
}Ans[155];

void insert(char *s,int Id)
{
    int len=strlen(s);
    int now=0,tmp;
    for(int i=0;i<len;++i)
    {
        tmp=s[i]-'a';
        if(nxt[now][tmp]==0)
            nxt[now][tmp]=++now_node;
        now=nxt[now][tmp];
    }
    id[now]=Id;
    return;
}

queue<int> que;
void getfail()
{
    for(int i=0;i<26;++i)
    {
        if(nxt[0][i])
        {
            que.push(nxt[0][i]);
        }
    }
    int now,tmp;
    while(!que.empty())
    {
        now=que.front(),que.pop();
        for(int i=0;i<26;++i)
        {
            tmp=nxt[now][i];
            if(tmp)
            {
                fail[tmp]=nxt[fail[now]][i];
                last[tmp]=id[fail[tmp]]?fail[tmp]:last[fail[tmp]];
                que.push(tmp);
            }
            else
                nxt[now][i]=nxt[fail[now]][i];
        }
    }
}

void query(char *s)
{
    int len=strlen(s);
    int now=0;
    for(int i=0;i<len;++i)
    {
        now=nxt[now][s[i]-'a'];
        for(int j=now;j;j=last[j])
            ++Ans[id[j]].cnt;
    }
}

int main()
{
    while(scanf("%d",&n)&&n!=0)
    {
        memset(nxt,0,sizeof(nxt));
        for(int i=0;i<=now_node;++i)
            fail[i]=0,last[i]=0,id[i]=0;
        now_node=0;
        for(int i=1;i<=n;++i)
        {
            scanf("%s",p[i]);
            insert(p[i],i);
            Ans[i].id=i,Ans[i].cnt=0;
        }
        Ans[0].cnt=0;
        getfail();
        scanf("%s",t);
        query(t);
        sort(Ans+1,Ans+n+1);
        cout<<Ans[1].cnt<<'\n';
        for(int i=1;i<=n;++i)
        {
            puts(p[Ans[i].id]);
            if(Ans[i].cnt!=Ans[i+1].cnt)
                break;
        }
    }
    return 0;
}