BZOJ3172[Tjoi2013]单词 题解

题目大意：

　　求一些字符串在一段文章中出现的次数。

思路：

　　AC自动机的经典应用，建完自动机直接将队列里的元素调Fail指针记录即可。

代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 using namespace std;
 #define M 1000009
 #define FC x==0?0:child[fail[x]][i]
 char s[M];
 int head,tail,cnt,q[M],sum[M],id[M],child[M][],fail[M];

 void ins(int &x)
 {
     for (int i=,j;i<strlen(s);i++,x=child[x][j],sum[x]++)
         if (!child[x][j=s[i]-'a']) child[x][j]=++cnt;
 }

 void build()
 {
     for (head=,tail=;head<tail;)
     {
         int x=q[++head],i;
         for (i=;i<;i++)
             if (child[x][i]) q[++tail]=child[x][i],fail[child[x][i]]=FC;
             else child[x][i]=FC;
     }
 }

 int main()
 {
     int n,i;
     scanf("%d",&n);
     for (i=;i<=n;i++) scanf("%s",s),ins(id[i]);
     for (build(),i=tail;i;i--) sum[fail[q[i]]]+=sum[q[i]];
     for (i=;i<=n;i++) printf("%d\n",sum[id[i]]);
 }