AC自动机模板

最新推荐文章于 2024-04-28 22:37:46 发布

原创最新推荐文章于 2024-04-28 22:37:46 发布 · 196 阅读

1 ·

CC 4.0 BY-SA版权

AC自动机专栏收录该内容

5 篇文章

订阅专栏

本文提供了一个AC自动机的实战代码实现，并详细解释了如何通过AC自动机进行字符串匹配的过程。代码中包括了节点结构定义、构建失败指针、主函数流程等关键部分，适用于对AC自动机原理及应用有一定了解的学习者。

#include <iostream>
#include <cstring>
#include <cstdio>
#include <algorithm>
#include <queue>
#include<ctime>
#define N 10000090
using namespace std;
int n;
int date[5000500][26],fail[N],cnt[N],tot=0,ans=0;
int st,pt;
char str[N],ptr[N];
queue<int> q;
void build()
{
    for(int i=0;i<26;i++)
        if(date[0][i])
           q.push(date[0][i]);
    while(!q.empty())
    {
        int re=q.front();q.pop();
        for(int i=0;i<26;i++)
        {
            if(!date[re][i])
            {
                date[re][i]=date[fail[re]][i];//直接跳转避免while
                continue;
            }
            q.push(date[re][i]);
            fail[date[re][i]]=date[fail[re]][i];
        }
    }
}
int main()
{
    double a=clock();
    freopen("in.in","r",stdin);
    freopen("my.out","w",stdout);
    scanf("%d",&n);
    while(n--)
    {
        scanf("%s",str+1);st=strlen(str+1);
        int j=0;
        for(int i=1;i<=st;i++)
        {
            if(!date[j][str[i]-'a']) date[j][str[i]-'a']=++tot;
            j=date[j][str[i]-'a'];
        }
        cnt[j]++;
    }
    build();
    scanf("%s",ptr+1);pt=strlen(ptr+1);
    int j=0;
    for(int i=1;i<=pt;i++)
    {
        j=date[j][ptr[i]-'a'];
        for(int t=j; t && cnt[t]!=-1;t=fail[t])
        {
            ans+=cnt[t];cnt[t]=-1;
        }
    }
    cout<<ans<<endl;
    cout<<clock()-a<<endl;
    return 0;
}