3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 3937 Solved: 1912
[Submit][Status][Discuss]
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
/* AC自动机模板 */ #include<iostream> #include<cstdio> #include<cstring> #define N 1000007 using namespace std; int sz,n;char s[N]; int son[N][26],fail[N],q[N],sum[N],loc[N]; inline void clear() { sz=1; for(int i=0;i<26;i++) son[0][i]=1; } void insert(int &pos) { int x=1,l=strlen(s); for(int i=0;i<l;i++) { if(!son[x][s[i]-'a']) son[x][s[i]-'a']=++sz; x=son[x][s[i]-'a'];sum[x]++;//以x为结尾的单词数加一 } pos=x; } void buildfail() { int head=0,tail=1;q[0]=1;fail[1]=0; while(head<tail) { int now=q[head++]; for(int i=0;i<26;i++)//更新fail指针 { if(!son[now][i]) continue; int ff=fail[now]; while(!son[ff][i]) ff=fail[ff]; fail[son[now][i]]=son[ff][i]; q[tail++]=son[now][i]; } } for(int i=tail-1;i>=0;i--) sum[fail[q[i]]]+=sum[q[i]];//因为以fail[q[i]]结尾的单词一定在以q[i]结尾的单词中出现 } int main() { clear(); scanf("%d",&n); for(int i=1;i<=n;i++) scanf("%s",s),insert(loc[i]); buildfail(); for(int i=1;i<=n;i++) printf("%d ",sum[loc[i]]); return 0; }