首页 > 代码库 > 3172: [Tjoi2013]单词
3172: [Tjoi2013]单词
3172: [Tjoi2013]单词
Time Limit: 10 Sec Memory Limit: 512 MBSubmit: 3246 Solved: 1565
[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
Source
在构建trie树时,统计每个节点出现的次数。
AC自动机实现:从前向后倒推,用一个单词的后缀的次数来更新前缀的次数。
#include<cstdio>#include<cstring>using namespace std;const int N=1e6+10;int n,cnt,h,t,sum[N],q[N],fail[N],w[201],tr[N][26];char s1[N];void ins(int &x){ int now=1,len=strlen(s1); for(int i=0;i<len;i++){ int t=s1[i]-‘a‘; if(!tr[now][t]) tr[now][t]=++cnt; now=tr[now][t];sum[now]++; } x=now;}void acmach(){ h=0;t=1; q[t]=1;fail[1]=0; while(h!=t){ int now=q[++h]; for(int i=0;i<26;i++){ if(!tr[now][i]) continue; int p=fail[now]; while(!tr[p][i]) p=fail[p]; fail[tr[now][i]]=tr[p][i]; q[++t]=tr[now][i]; } }}void solve(){ for(int i=t;i;i--) sum[fail[q[i]]]+=sum[q[i]]; for(int i=1;i<=n;i++) printf("%d\n",sum[w[i]]);}int main(){ scanf("%d",&n);cnt=1; for(int i=0;i<26;i++) tr[0][i]=1; for(int i=1;i<=n;i++) scanf("%s",s1),ins(w[i]); acmach();solve(); return 0;}
3172: [Tjoi2013]单词
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。