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][26],fail[M];

void ins(int &x)
{
    for (int i=0,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=0,tail=1;head<tail;)
    {
        int x=q[++head],i;
        for (i=0;i<26;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=1;i<=n;i++) scanf("%s",s),ins(id[i]);
    for (build(),i=tail;i;i--) sum[fail[q[i]]]+=sum[q[i]];
    for (i=1;i<=n;i++) printf("%d\n",sum[id[i]]);
}