洛谷P5357 AC自动机模板题

P5357
对应输出n个模式串在文本串中出现的次数
AC自动机😳
坑点：可能出现重复模板串，所以标记结尾时做相应的修改即可
这里的写法是把fail树对应的图建出来了，
其实不需要建也可以求出(这不重要)

void getans(void){
    for(int i=1;i<=tot;i++) in[fail[i]]++;
    for(int i=1;i<=tot;i++) if(!in[i])
         q.push(i);
    while(!q.empty()){
        int x=q.front();
        q.pop();
        ans[fail[x]]+=ans[x];
        in[fail[x]]--;
        if(!in[fail[x]])
            q.push(fail[x]);
    }
}

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
typedef pair<int,int>pii;
typedef pair<LL,LL>pll;
const int maxn=2e6+5;
const int mod=1e9+7;
const int oo=1e9+8;
char s[maxn+5];
int fail[maxn+5],trie[maxn+5][26],tot=0,ans[maxn+5],dp[maxn+5];
int n;
queue<int>q;
vector<int>vec[maxn+5],ed[maxn+5];
void dfs(int x)
{
    int len=vec[x].size();
    for(int i=0;i<len;i++){
        int y=vec[x][i];
        dfs(y);
        dp[x]+=dp[y];
    }
    for(int i=0;i<ed[x].size();i++)
        ans[ed[x][i]]=dp[x];
}
struct Ac_auto
{
    void add(char *s,int id){
        int len=strlen(s),p=0;
        for(int i=0;i<len;i++){
            int c=s[i]-'a';
            if(!trie[p][c]) trie[p][c]=++tot;
            p=trie[p][c];
        }
        ed[p].push_back(id);
    }
    void build_fail(){
        while(!q.empty()) q.pop();
        for(int i=0;i<26;i++)
            if(trie[0][i]) q.push(trie[0][i]),vec[0].push_back(trie[0][i]);

        while(!q.empty()){
            int f=q.front();q.pop();

            for(int i=0;i<26;i++){
                if(trie[f][i]){
                    fail[trie[f][i]]=trie[fail[f]][i];
                    vec[fail[trie[f][i]]].push_back(trie[f][i]);
                    q.push(trie[f][i]);
                }
                else trie[f][i]=trie[fail[f]][i];
            }
        }
    }
    void check(char *s){
        for(int i=1;i<=n;i++) ans[i]=0,dp[i]=0;
        int p=0,len=strlen(s);
        for(int i=0;i<len;i++){
             p=trie[p][s[i]-'a'];
             dp[p]++;
        }
        dfs(0);
    }
}ac;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%s",s);
        ac.add(s,i);
    }
    scanf("%s",s);

    ac.build_fail();
    ac.check(s);
    for(int i=1;i<=n;i++){
        printf("%d\n",ans[i]);
    }
}