bzoj 3172 [Tjoi2013]单词

3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 3852 Solved: 1864
[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

---

【分析】
• 构建单词的AC⾃自动机
• ⽂文章在⾃自动机上跑⼀一遍
• 在每个状态上打标记
• 对每个单词，在fail树的⼦子树中求和即可

---

【代码】

//bzoj 3172 [Tjoi2013]单词#include<queue>#include<cstdio>#include<cstring>#include<iostream>#include<algorithm>#define ll long long#define fo(i,j,k) for(int i=j;i<=k;i++)using namespace std;const int mxn=1000010;queue <int> q;char s[mxn];int w[mxn],wxl[mxn],n,num,tot;struct node{    int son[30];    int fail,cnt;}a[mxn];inline void clear(int x){    memset(a[x].son,0,sizeof a[x].son);}inline void trie(int u){    int x=0,len=strlen(s);    fo(i,0,len-1)    {        int t=s[i]-'a'+1;        if(!a[x].son[t])        {            num++;            clear(num);            a[x].son[t]=num;        }        x=a[x].son[t];        a[x].cnt++;    }    w[u]=x;}inline void build(){    fo(i,1,26)      if(a[0].son[i])        q.push(a[0].son[i]);    while(!q.empty())    {        int x=q.front();        wxl[++tot]=x;        q.pop();        fo(i,1,26)        {            int y=a[x].son[i],fail=a[x].fail;            if(y)              a[y].fail=a[fail].son[i],q.push(y);            else              a[x].son[i]=a[fail].son[i];        }    }}int main(){    scanf("%d",&n);    fo(i,1,n)    {        scanf("%s",s);        trie(i);    }    build();    for(int i=tot;i;i--)      a[a[wxl[i]].fail].cnt+=a[wxl[i]].cnt;    fo(i,1,n) printf("%d\n",a[w[i]].cnt);    return 0;}