3172: [Tjoi2013]单词

Posted 2020-08-26 神犇(shenben)

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