[bzoj3172][AC自动机]单词

Description

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

题解

最近学了AC自动机就专门向栋爷爷的blog要了一题做做。。
结果发现题意说的不清不楚。大概就是每个单词在所有单词中出现的次数吧
可以后缀数组做，但是AC自动机可能好理解点
对单词建一棵Tire树，然后建失败指针
所有失败指针连起来，可以连成一棵树。
对于节点i及节点i的fail指针j
root~j这一段，在root~i这里一定有出现过。那么j的s就要加上i的s，s是在建Tire时处理的，表示root~当前节点的串总共出现了多少次
然后for一下每个单词的末尾，输出s就行啦

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
struct Tire
{
    int c[27],s,fail;
    Tire(){memset(c,-1,sizeof(c));s=fail=0;}
}tr[511000];int tot;
char s[1110000];
int ed[210];
void add(int id,int root)
{
    int x=root,len=strlen(s+1);
    for(int i=1;i<=len;i++)
    {
        int j=s[i]-'a'+1;
        if(tr[x].c[j]==-1)tr[x].c[j]=++tot;
        x=tr[x].c[j];
        tr[x].s++;
    }
    ed[id]=x;
}
int list[1110000];
void build_fail()
{
    int head=1,tail=1;
    list[1]=0;
    while(head<=tail)
    {
        int x=list[head];
        for(int i=1;i<=26;i++)
        {
            int son=tr[x].c[i];
            if(son==-1)continue;
            if(x==0)tr[son].fail=0;
            else
            {
                int j=tr[x].fail;
                while(j && tr[j].c[i]==-1)j=tr[j].fail;
                tr[son].fail=max(tr[j].c[i],0);
            }
            list[++tail]=son;
        }
        head++;
    }
    //对于每一个fail节点，在i上一定有一段与之相同的。那么在i上出现了的，在fail上也一定出现，就加起来 
    for(int i=tail;i;i--)tr[tr[list[i]].fail].s+=tr[list[i]].s;
}
int c[210];
int n;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
    {
        scanf("%s",s+1);
        add(i,0);
    }
    build_fail();
    for(int i=1;i<=n;i++)printf("%d\n",tr[ed[i]].s);
    return 0;
}