LG5357 「模板」AC自动机（二次加强版） AC自动机+fail树

问题描述

LG5357

题解

不是fail树的AC自动机复杂度是假的。

把AC自动机搞出来，建立Trie树，树上爆搜一遍就好了。

\(\mathrm{Code}\)

#include<bits/stdc++.h>
using namespace std;

template <typename Tp>
void read(Tp &x){
    x=0;char ch=1;int fh;
    while(ch!='-'&&(ch>'9'||ch<'0')) ch=getchar();
    if(ch=='-') ch=getchar(),fh=-1;
    else fh=1;
    while(ch>='0'&&ch<='9') x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
    x*=fh;
}

int root,ch[200007][26],tot,n;
char s[2000003];

int pos[200007],val[200007],fail[200007];
int m,ed[200007];

int chk(char s){
    return s-'a';
}

void insert(int x){
    int p=root;
    for(int i=1;i<=n;i++){
        int k=chk(s[i]);
        if(!ch[p][k]) ch[p][k]=++tot;
        p=ch[p][k];
    }
    ++ed[p],pos[x]=p;
}

void query(){
    int p=root;
    for(int i=1;i<=n;i++){
        p=ch[p][chk(s[i])];
        ++val[p];
    }
}

void build(){
    queue<int>q;
    for(int i=0;i<26;i++){
        if(ch[root][i]) q.push(ch[root][i]);
    }
    while(!q.empty()){
        int x=q.front();q.pop();
        for(int i=0;i<26;i++){
            if(ch[x][i]) fail[ch[x][i]]=ch[fail[x]][i],q.push(ch[x][i]);
            else ch[x][i]=ch[fail[x]][i];
        }
    }
}

int Head[200007],Next[200007],cnt,to[200007];

void add(int x,int y){
    to[++cnt]=y,Next[cnt]=Head[x],Head[x]=cnt;
}

int size[200007];

void dfs(int x){
    size[x]=val[x];
    for(int i=Head[x];i;i=Next[i]){
        dfs(to[i]);size[x]+=size[to[i]];
    }
}

int main(){
    ios::sync_with_stdio(0);
    cin>>m;
    for(int i=1;i<=m;i++){
        cin>>(s+1);n=strlen(s+1);
        insert(i);
    }
    cin>>(s+1);n=strlen(s+1);
    build();query();
    for(int i=1;i<=tot;i++) add(fail[i],i);
    dfs(root);
    for(int i=1;i<=m;i++) cout<<size[pos[i]]<<endl;
    return 0;
}