P3966 [TJOI2013]单词

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
using namespace std;
struct data{
    int nxt[26],end,fail,last;
}a[1000002];
int n,cnt,pre[202];
long long f[202];
string q,g;
inline void Trie_build(int id){
    cin>>q; g=g+"#"+q; //用string方便地加上各个单词
    int u=0,len=q.size();
    for(int i=0;i<len;++i){
        int p=q[i]-‘a‘;
        if(!a[u].nxt[p]) a[u].nxt[p]=++cnt;
        u=a[u].nxt[p];
    }
    if(!a[u].end) a[u].end=id;
    pre[id]=a[u].end; //重复单词处理
}
inline void AC_build(){
    queue <int> h;
    for(int i=0;i<26;++i) if(a[0].nxt[i]) h.push(a[0].nxt[i]);
    while(!h.empty()){
        int x=h.front(); h.pop();
        for(int i=0;i<26;++i){
            int &to=a[x].nxt[i];
            if(to){
                a[to].fail=a[a[x].fail].nxt[i];
                a[to].last= a[a[to].fail].end ? a[to].fail:a[a[to].fail].last; //last指针优化
                h.push(to);
            }else to=a[a[x].fail].nxt[i];
        }
    }
}
inline void AC_query(){
    int u=0,len=g.size();
    for(int i=0;i<len;++i){
        if(g[i]==‘#‘) {u=0; continue;}
        u=a[u].nxt[g[i]-‘a‘];
        for(int j=u;j;j=a[j].last) ++f[a[j].end]; //按last指针跳转
    }
    for(int i=1;i<=n;++i) printf("%lld\n",f[pre[i]]); //输出的是 指向的节点 的编号 的值
}
int main(){
    //freopen("P3966.in","r",stdin);
    scanf("%d",&n);
    for(int i=1;i<=n;++i) Trie_build(i);
    AC_build();
    AC_query();
    return 0;
}