3172: [Tjoi2013]单词

3172: [Tjoi2013]单词

Time Limit: 10 Sec Memory Limit: 512 MB
Submit: 2783 Solved: 1322
[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

#include<iostream>
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#include<math.h>
#include<algorithm>
#include<queue>
using namespace std;
#define LL long long
int read()
{
    int s=0,f=1;char ch=getchar();
    while(!('0'<=ch&&ch<='9')){if(ch=='-')f=-1;ch=getchar();}
    while('0'<=ch&&ch<='9'){s=(s<<3)+(s<<1)+ch-'0';ch=getchar();}
    return s*f;
}
int n,np;
int ch[1000006][26],fail[1000006];
char z[1000006];
queue<int>q;
int siz[1000006];
int bfn[1000006];
int end[205];
int main()
{
    //freopen(".in","r",stdin);
    //freopen(".out","w",stdout);
    n=read();
    for(int i=1;i<=n;i++)
       {scanf("%s",z+1);
        int l=strlen(z+1),x=0;
        for(int j=1;j<=l;j++)
           {if(!ch[x][z[j]-'a'])
                ch[x][z[j]-'a']=++np;
            x=ch[x][z[j]-'a'];
            siz[x]++;
           }
        end[i]=x;
       }
    for(q.push(0);!q.empty();)
       {int u=q.front();q.pop();
        bfn[++bfn[0]]=u;
        for(int i=0;i<26;i++)
           {int &v=ch[u][i];
            if(!v)
               {v=ch[fail[u]][i];
                continue;
               }
            if(u)fail[v]=ch[fail[u]][i];
            q.push(v);
           }
       }
    for(int i=bfn[0];i;i--)
        siz[fail[bfn[i]]]+=siz[bfn[i]];
    for(int i=1;i<=n;i++)
        printf("%d\n",siz[end[i]]);
    //fclose(stdin);
    //fclose(stdout);
    return 0;
}