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;
}