BZOJ3172:[TJOI2013]单词(AC自动机)

Description

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

Input

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

Output

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

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

Solution

还是对AC自动机的理解不够到位啊=-=……
其实fail指针也是一颗树，而fail树有一些神奇的性质
根节点仍然是0，每一个fail指针指向自己的最长后缀
也就是说，我们把每个节点的权值设为1，然后再处理一下fail树的子树和
如何处理呢？在build_fail的时候记录一下经过节点的顺序，
然后倒序枚举，将当前枚举到的点的fail指针指向的点的权值累加一下
最开始记录一下每个单词的位置，方便输出。

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<queue>
#define N (1000000+100)
using namespace std;
int Son[N][27],Fail[N],End[N];
int n,sz,ans[N],pos[N],Q[N],cnt;
bool vis[N],flag;
char s[N],ask[N];
queue<int>q;
void Insert(char s[],int &pos)
{
    int now=0,len=strlen(s);
    for (int i=0; i<len; ++i)
    {
        int x=s[i]-'a';
        if (!Son[now][x]) Son[now][x]=++sz;
        now=Son[now][x];
        ++ans[now];
    }
    pos=now;
}

void Build_Fail()
{
    for (int i=0;i<26;++i)
        if (Son[0][i])
            q.push(Son[0][i]);
    while (!q.empty())
    {
        int now=q.front(); q.pop();
        Q[++cnt]=now;
        for (int i=0;i<26;++i)
        {
            if (!Son[now][i])
            {
                Son[now][i]=Son[Fail[now]][i];
                continue;
            }
            Fail[Son[now][i]]=Son[Fail[now]][i];
            q.push(Son[now][i]);
        }
    }    
}

void Query()
{
    for (int i=cnt;i>=1;--i)
        ans[Fail[Q[i]]]+=ans[Q[i]];
    for (int i=1;i<=n;++i)
        printf("%d\n",ans[pos[i]]);
}
int main()
{
    scanf("%d",&n);
    for (int i=1;i<=n;++i)
        scanf("%s",s),Insert(s,pos[i]);
    Build_Fail();
    Query();
}