BZOJ_P3172 [Tjoi2013]单词(AC自动机)

BZOJ传送门

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 2496 Solved: 1163
[Submit][Status][Discuss]
Description
某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

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

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

Sample Input
3
a
aa
aaa

Sample Output
6
3
1

HINT

Source

#include<cstdio>
#include<queue>
#include<iostream>
using namespace std;
#define M 1000005
#define N 205
struct AC{
    struct Node{int f,cnt;int ch[26];}d[M];
    int n,cnt,h,t;int w[N],q[M];

    void build(){
        q[1]=1;d[1].f=1;
        for(h=0,t=1;h<t;){
            int x=q[++h];
            for(int i=0;i<26;++i) {
                if(!d[x].ch[i]){d[x].ch[i]=d[d[x].f].ch[i];continue;}
                int u=d[x].ch[i],v=d[x].f;
                q[++t]=u;
                d[u].f=d[v].ch[i]&&d[v].ch[i]!=u?d[v].ch[i]:1;
            }
        }
    }

    void init(){
        scanf("%d",&n);char ch=getchar();cnt=1;
        for(int i=1,tmp;i<=n;i++){
            while(ch>'z'||ch<'a') ch=getchar();
            for(tmp=1;ch>='a'&&ch<='z';ch=getchar()){
                if(!d[tmp].ch[ch-'a']) d[tmp].ch[ch-'a']=++cnt;
                ++d[tmp=d[tmp].ch[ch-'a']].cnt; 
            } 
            w[i]=tmp;
        }
        build();
        for(int i=t;i;--i)  d[d[q[i]].f].cnt+=d[q[i]].cnt;
        for(int i=1;i<=n;i++) printf("%d\n",d[w[i]].cnt);
    }
}ac;
int main(){
    ac.init();
    return 0;
}