首页 > 代码库 > 3172: [Tjoi2013]单词

3172: [Tjoi2013]单词

3172: [Tjoi2013]单词

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

在构建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]单词