luogu_3966【题解】单词 AC自动机

#include<bits/stdc++.h>
#define sc(x) scanf("%d",&x)
using namespace std;
const int maxn=1e6+10;
int n,cnt;
struct trie{
    int fail,num;
    int ch[30];
    #define fa(x) ac[x].fail
    #define nu(x) ac[x].num
}ac[maxn];
int a[maxn],h[maxn];
inline void insert(string s,int x){
    int len=s.length();
    int now=0;
    for(int i=0;i<len;i++){
        if(!ac[now].ch[s[i]-‘a‘])
            ac[now].ch[s[i]-‘a‘]=++cnt;
        now=ac[now].ch[s[i]-‘a‘];
        nu(now)++;
    }
    a[x]=now;
}
//bfs
inline void getfail(){
    int head=0;int tail=0;
    for(int i=0;i<26;i++)
        if(ac[0].ch[i])
            fa(ac[0].ch[i])=0,h[++tail]=ac[0].ch[i];
    while(head<tail){
        int now=h[++head];
        for(int i=0;i<26;i++){
            if(ac[now].ch[i]){
                fa(ac[now].ch[i])=ac[fa(now)].ch[i];
                h[++tail]=ac[now].ch[i];
            }
            else
                ac[now].ch[i]=ac[fa(now)].ch[i];
        }
    }
} 
inline void ask(){
    for(int i=cnt;i>=0;i--) nu(fa(h[i]))+=nu(h[i]);
    for(int i=1;i<=n;i++) printf("%d\n",nu(a[i]));
}
int main()
{
    sc(n);
    string s;
    for(int i=1;i<=n;i++){
        cin>>s;
        insert(s,i);
    }
    fa(0)=0;
    getfail();
    ask();
    // while(1);
    return 0;
}