P5357 【模板】AC自动机（二次加强版）

后缀自动机与字典树详解

最新推荐文章于 2022-11-21 02:09:30 发布

原创最新推荐文章于 2022-11-21 02:09:30 发布 · 430 阅读

1 ·

CC 4.0 BY-SA版权

ACM题解专栏收录该内容

112 篇文章

订阅专栏

本文深入探讨了后缀自动机和字典树的实现与应用，通过具体的编程示例，详细讲解了如何使用这两种数据结构解决字符串匹配问题。文章提供了完整的代码示例，并介绍了如何构建字典树，以及如何通过DFS遍历fail树来解决问题。

https://www.luogu.org/problemnew/show/P5357

题意很简单。虽然说是模板，可能是后缀自动机的模板吧！！还不会

感觉和这道题差不多链接题解

但不过知道上道题怎么这道题应该知道会怎么做。

承接上一道题，上一道题是所有的模式串组成文本串；

所有这道题只需要用文本串去跑一个字典树，然后记录一个cnt数组，然后一个dfs遍历fail树就完了，和上一道题一样的。

#include<iostream>
#include<string.h>
#include<algorithm>
#include<stdio.h>
#include<math.h>
#include<stdlib.h>
#include<queue>
using namespace std;
typedef long long ll;
const int N=2e5+10;
int nxt[N][26],endd[N],fail[N];
int root,l;

void Insert(string s,int k)
{
    int len=s.size();
    int u=root;
    for(int i=0;i<len;i++)
    {
        int id=s[i]-'a';
        if(nxt[u][id]==0) nxt[u][id]=++l;
        u=nxt[u][id];
    }
    endd[k]=u;
}

void build()
{
    queue<int>qu;
    for(int i=0;i<26;i++)
        if(nxt[root][i]) qu.push(nxt[root][i]);
    while(!qu.empty())
    {
        int u=qu.front();qu.pop();
        ///endd[u]+=endd[fail[u]];
        for(int i=0;i<26;i++)
        {
            if(nxt[u][i]) fail[nxt[u][i]]=nxt[fail[u]][i],qu.push(nxt[u][i]);
            else nxt[u][i]=nxt[fail[u]][i];
        }
    }
}
int cnt[N*10];
vector<int>vec[N];
void dfs(int u)
{
    for(int i=0;i<vec[u].size();i++)
    {
        int v=vec[u][i];
        dfs(v);
        cnt[u]+=cnt[v];
    }
}

string str[N];
char s[N*10];
int main()
{
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        cin>>str[i];
        Insert(str[i],i);
    }
    build();
    scanf("%s",s);
    int len=strlen(s),p=0;
    for(int i=0;i<len;i++)
    {
        p=nxt[p][s[i]-'a'];
        cnt[p]++;
    }
    for(int i=1;i<=l;i++) vec[fail[i]].push_back(i);
    dfs(0);
    for(int i=1;i<=n;i++) printf("%d\n",cnt[endd[i]]);
    return 0;
}