bzoj3172 [Tjoi2013]单词（AC自动机+fail树）

求每个串在文章中出现的次数（文章为这些串加分隔符串联而成）
考虑ACAM的朴素做法，就是建出ACAM以后每个串在ACAM跑一遍，跳fail指针统计答案。但是这样最后复杂度是$O(n^2)$的。会TLE
怎么办呢？我们考虑建出fail树，那么我们每次在节点p跳fail指针就相当于在fail树上从p一直跳到根，给路径上所有节点的cnt+1。于是我们可以每次只给cnt[p]+1,然后节点p的答案就是他的子树权值和，可以一遍O(n)处理出每个点的答案。

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <queue>
using namespace std;
#define ll long long
#define N 1000010
#define inf 0x3f3f3f3f
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int m,n=0,rt,id[210],son[N][26],fail[N],cnt[N],q[N];
char s[N];
inline void ins(char *s,int ii){
    int len=strlen(s+1),p=rt;
    for(int i=1;i<=len;++i){
        int &y=son[p][s[i]-'a'];if(!y) y=++n;p=y;cnt[p]++;
    }id[ii]=p;
}
inline void buildAC(){
    int qh=1,qt=0;q[++qt]=rt;
    while(qh<=qt){
        int x=q[qh++];
        for(int i=0;i<26;++i){
            int &y=son[x][i];
            if(!y){y=son[fail[x]][i];continue;}q[++qt]=y;
            fail[y]=son[fail[x]][i];
        }
    }for(int i=qt;i>=1;--i) cnt[fail[q[i]]]+=cnt[q[i]];
}
int main(){
//  freopen("a.in","r",stdin);
    m=read();rt=++n;
    for(int i=0;i<26;++i) son[0][i]=rt;
    for(int i=1;i<=m;++i) scanf("%s",s+1),ins(s,i);buildAC();
    for(int i=1;i<=m;++i) printf("%d\n",cnt[id[i]]);
    return 0;
}