P5357 【模板】AC 自动机

题目背景

本题原为“AC 自动机（二次加强版）”。完成本题前可以先完成 AC 自动机（简单版） 和 AC 自动机（简单版 II） 两道题，为 AC 自动机更简单的应用。

题目描述

给你一个文本串 \(S\) 和 \(n\) 个模式串 \(T_{1 \sim n}\)，请你分别求出每个模式串 \(T_i\) 在 \(S\) 中出现的次数。

输入格式

第一行包含一个正整数 \(n\) 表示模式串的个数。

接下来 \(n\) 行，第 \(i\) 行包含一个由小写英文字母构成的非空字符串 \(T_i\)。

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

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

输出格式

输出包含 \(n\) 行，其中第 \(i\) 行包含一个非负整数表示 \(T_i\) 在 \(S\) 中出现的次数。

输入输出样例 #1

输入 #1

5
a
bb
aa
abaa
abaaa
abaaabaa

输出 #1

6
0
3
2
1

说明/提示

对于 \(100 \%\) 的数据，\(1 \le n \le 2 \times {10}^5\)，\(T_{1 \sim n}\) 的长度总和不超过 \(2 \times {10}^5\)，\(S\) 的长度不超过 \(2 \times {10}^6\)。

把这个数据加强版的代码稍加修改即可通过另外两题了

题解

#include <bits/stdc++.h>
using namespace std;
const int N=2e5+10;
typedef long long ll;
int n;
int tree[N][26];
int fail[N];
int cnt = 0;
int ed[N];
int t[N];
vector<int>e[N];

void insert(int idx,string s)
{
    int u = 0;
    for(auto ch:s)
    {
        int c = ch - 'a';
        if(tree[u][c]==0)tree[u][c] = ++cnt;
        u = tree[u][c];
    }
    ed[idx] = u;
} 

void setfail()
{
    queue<int>q;
    for(int c =0;c<26;++c)
    {
        if(tree[0][c])q.push(tree[0][c]);
    }
    while(!q.empty())
    {
        int u = q.front();q.pop();
        for(int c=0;c<26;c++)
        {
            if(tree[u][c]==0)
            {
                tree[u][c] = tree[fail[u]][c];
            }
            else
            {
                fail[tree[u][c]] = tree[fail[u]][c];
                q.push(tree[u][c]);
            }
        }
    }
}

void f1(int u)
{
    for(int v:e[u])
    {
        f1(v);
        t[u]+=t[v];
    }
}

void solve()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        string s;
        cin>>s;
        insert(i,s);
    }

    setfail();
    string text;
    cin>>text;
    int u = 0;
    for(auto ch:text)
    {
        int c = ch-'a';
        u = tree[u][c];
        t[u]++;
    }

    for(int i=1;i<=cnt;i++)
    {
        e[fail[i]].push_back(i);
    }

    f1(0);

    for(int i=1;i<=n;i++)
    {
        cout<<t[ed[i]]<<endl;
    }
}
int main()
{
    ios::sync_with_stdio(0);
    cin.tie(0);
    cout.tie(0);
    int _=1;
    // cin>>_;
    while(_--)
    {
        solve();

    }
    return 0;
}