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

最新推荐文章于 2022-08-30 12:42:16 发布

0MagiCian

最新推荐文章于 2022-08-30 12:42:16 发布

阅读量172

点赞数 1

文章标签： AC自动机字符串匹配模式查找算法实现 C++

本文链接：https://blog.csdn.net/qq_51518266/article/details/123405242

版权

【模板】AC 自动机（二次加强版） - 洛谷

题目描述

给你一个文本串 SS 和 nn 个模式串 T_{1 \sim n}T1∼n，请你分别求出每个模式串 T_iTi 在 SS 中出现的次数。

输入格式

第一行包含一个正整数 nn 表示模式串的个数。

接下来 nn 行，第 ii 行包含一个由小写英文字母构成的非空字符串 T_iTi。

最后一行包含一个由小写英文字母构成的非空字符串 SS。

数据不保证任意两个模式串不相同。

输出格式

输出包含 nn 行，其中第 ii 行包含一个非负整数表示 T_iTi 在 SS 中出现的次数。

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<queue>
using namespace std;
const int N1=2e5+5;
int in[N1];
int vis[N1];
int cnt;
int MAP[N1];
int n;
char str[2000006];
struct ACautomata{
    int tire[N1][26];
    int fail[N1];
    int ans[N1];
    int flag[N1];
    void insert(char *s,int id)
    {
        int len=strlen(s);
        int p=0;
        for(int i=0;i<len;i++)
        {
            if(!tire[p][s[i]-'a'])tire[p][s[i]-'a']=++cnt;
            p=tire[p][s[i]-'a'];
        }
        if(!flag[p])flag[p]=id;
        MAP[id]=flag[p];
    }
    void build()
    {
        queue<int> que;
        for(int i=0;i<26;i++)if(tire[0][i])que.push(tire[0][i]);
        while(!que.empty())
        {
            int v=que.front();que.pop();
            for(int i=0;i<26;i++)
            {
                if(tire[v][i]){
                    fail[tire[v][i]]=tire[fail[v]][i];
                    in[fail[tire[v][i]]]++;
                    que.push(tire[v][i]);
                }
                else tire[v][i]=tire[fail[v]][i];
            }
        }
    }
    void query(char * s)
    {
        int len=strlen(s);
        int p=0;
        for(int i=0;i<len;i++)
        {
            p=tire[p][s[i]-'a'];
            ans[p]++;
        }
    }
}ac;
void topu()
{
    queue<int> que;
    for(int i=1;i<=cnt;i++)if(in[i]==0)que.push(i);
    while(!que.empty())
    {
        int v=que.front();que.pop();
        vis[ac.flag[v]]=ac.ans[v];
        ac.ans[ac.fail[v]]+=ac.ans[v];
        in[ac.fail[v]]--;
        if(in[ac.fail[v]]==0)que.push(ac.fail[v]);
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",str);
        ac.insert(str,i);
    }
    ac.build();
    scanf("%s",str);
    ac.query(str);
    topu();
    for(int i=1;i<=n;i++)
    {
        printf("%d\n",vis[MAP[i]]);
    }
    return 0;
}