Description
某人读论文,一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次,现在想知道每个单词分别在论文中出现多少次。
Input
第一个一个整数N,表示有多少个单词,接下来N行每行一个单词。每个单词由小写字母组成,N<=200,单词长度不超过10^6
Output
输出N个整数,第i行的数字表示第i个单词在文章中出现了多少次。
Sample Input
3
a
aa
aaa
a
aa
aaa
Sample Output
6
3
1
3
1
HINT
Source
开心~又在bzoj找到一题水题。。
注意一下题目题意有点点不清晰。。。
其实总的串是N个单词全部连接起来,而且每两个之间有一个空格(也可以用其它字符代替)
还有就是是N个单词全部连起来的长度<=10^6……
题目比较良心的地方是内存开了512M,这样子的话可以直接上trie[][]了。
总之是一道水水的模板题吧。。
#include<bits/stdc++.h>
using namespace std;
const int
N=1000205,
NUM=205;
int n,cnt,LenAll;
int pp[N];
int Ans[NUM],fail[N],num[N],Q[N];
int trie[N][27];
char All[N],tmp[N];
int find(int x,int t){
if (trie[x][t]) return trie[x][t];
if (!x) return 0;
return find(fail[x],t);
}
void insert(int ii){
int len=strlen(tmp),now=0;
for (int i=0;i<len;i++){
int j=tmp[i]-'a';
if (trie[now][j]) now=trie[now][j];
else now=trie[now][j]=++cnt;
}
if (num[now]) pp[ii]=num[now];
else num[now]=ii;
}
void AC(){
int head=0,tail=1;
Q[0]=0;
while (head!=tail){
int now=Q[head++];
for (int i=0;i<27;i++){
if (!trie[now][i]) continue;
int t=find(fail[now],i);
if (!now) t=0;
fail[trie[now][i]]=t;
Q[tail++]=trie[now][i];
}
}
}
void solve(){
int now=0;
for (int i=0;i<LenAll;i++){
int j=All[i]-'a';
now=find(now,j);
for (int j=now;j;j=fail[j])
if (num[j]) Ans[num[j]]++;
}
}
int main(){
scanf("%d",&n);
LenAll=cnt=0;
for (int i=1;i<=n;i++){
scanf("%s",tmp);
int len=strlen(tmp);
for (int j=0;j<len;j++) All[LenAll++]=tmp[j];
All[LenAll++]='{';
insert(i);
}
AC();
solve();
for (int i=1;i<=n;i++)
if (pp[i]) Ans[i]=Ans[pp[i]];
for (int i=1;i<=n;i++)
printf("%d\n",Ans[i]);
return 0;
}