AC自动机（简单版）--P3808 【模板】

最新推荐文章于 2024-11-12 16:24:06 发布

原创最新推荐文章于 2024-11-12 16:24:06 发布 · 225 阅读

0 ·

CC 4.0 BY-SA版权

题目总结专栏收录该内容

43 篇文章

订阅专栏

博客围绕一个问题展开，即给出n个模式串和一个文本串，求文本串中出现的模式串数量。这是AC自动机裸题，作者看过多个版本的AC自动机后理解了代码实现，也体会到懂原理与用原理解决问题间的差距。

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

题意：给出n个模式串和一个文本串，问在文本串中出现了几个模式串。

思路：AC自动机裸题；

看过n个版本的AC自动机后终于理解了代码是如何实现的。再一次体会到光懂得原理和能利用原理解决问题之间的巨大的鸿沟。

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <vector>
#include <queue>
#define INF 0x3f3f3f3f
#define FRE() freopen("in.txt","r",stdin)
using namespace std;
const int maxn = 1e6+10;
typedef long long ll;
typedef pair<int,int> P;
int cnt = 0;
struct AC
{
    int next[26];
    int en,fail;
}ac[maxn];

void Build(char* str)//建字典树
{
    int rt = 0;
    for(int i = 0; str[i]; i++)
    {
        int index = str[i] - 'a';
        if(!ac[rt].next[index])
            ac[rt].next[index] = ++cnt;
        rt = ac[rt].next[index];
    }
    ac[rt].en++;
}

void getFail()//获取fail指针
{
    queue<int> que;
    for(int i = 0; i < 26; i++)
    {
        if(ac[0].next[i])
        {
            ac[ac[0].next[i]].fail = 0;//将根节点孩子的fail指针指向根，即0
            que.push(ac[0].next[i]);
        }
    }
    while(!que.empty())
    {
        int now = que.front(); que.pop();
        for(int i = 0; i < 26; i++)
        {
            if(!ac[now].next[i])//如果该节点孩子没有出现过，就将这个节点的孩子等于该节点fail指向的节点的孩子
                ac[now].next[i] = ac[ac[now].fail].next[i];
            else//出现过就将这个节点孩子的fail指向这个节点的fail指向的节点的孩子
            {
                ac[ac[now].next[i]].fail = ac[ac[now].fail].next[i];
                que.push(ac[now].next[i]);
            }
        }
    }
}

int Fun(char* str)
{
    int rt = 0,res = 0;
    for(int i = 0; str[i]; i++)
    {
        rt = ac[rt].next[str[i]-'a'];
        for(int j = rt; j&&ac[j].en!=-1; j= ac[j].fail)
        {
            res += ac[j].en;
            ac[j].en = -1;
        }
    }
    return res;
}

char buf[maxn];
int main()
{

    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%s",buf);
        Build(buf);
    }
    getFail();
    scanf("%s",buf);
    printf("%d\n",Fun(buf));
    return 0;
}