3172: [Tjoi2013]单词

Description

某人读论文，一篇论文是由许多单词组成。但他发现一个单词会在论文中出现很多次，现在想知道每个单词分别在论文中出现多少次。

Input

第一个一个整数N,表示有多少个单词，接下来N行每行一个单词。每个单词由小写字母组成，N<=200,单词长度不超过10^6

Output

输出N个整数，第i行的数字表示第i个单词在文章中出现了多少次。

Sample Input

3
a
aa
aaa

Sample Output

6
3
1

 

 

这道题就是给定n个字符串，询问每个字符串在其他的字符串中出现了几次。。。

很容想到可以用AC自动机来做，我们对于每个字符串的每个节点都沿着fail指针走，每走到一个节点就把这个节点++；

但是这样做会超时。。。

引入fail树，fail树就是把AC自动机的fail指针反向，就构成了一棵树，我们在树上走就不会T了。。。。

#include<iostream>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<string>
#include<map>
#include<queue>
#include<vector>
#include<set>
#define inf 1000000000
#define maxn 1000000+5
#define maxm 26
#define eps 1e-10
#define ll long long
#define for0(i,n) for(int i=0;i<=(n);i++)
#define for1(i,n) for(int i=1;i<=(n);i++)
#define for2(i,x,y) for(int i=(x);i<=(y);i++)
#define for3(i,x,y) for(int i=(x);i>=(y);i--)
#define for4(i,x) for(int i=head[x],y=e[i].go;i;i=e[i].next,y=e[i].go)
using namespace std;
int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=10*x+ch-'0';ch=getchar();}
    return x*f;
}
int n,pos[maxn];
struct acm{
    int tot;
    int next[maxn][maxm],sum[maxn],fail[maxn],q[maxn];
    char ch[maxn];
    acm(){
        tot=1;
        for0(i,25)next[0][i]=1;
    }
    void insert(int &pos){
        int now=1;
        scanf("%s",ch);
        int n=strlen(ch);
        for0(i,n-1){
            if(!next[now][ch[i]-'a'])next[now][ch[i]-'a']=++tot;
            now=next[now][ch[i]-'a'];
            sum[now]++;
        }
        pos=now;
    }
    void build(){
        int head=0,tail=1;
        q[0]=1;fail[1]=0;
        while(head!=tail){
            int now=q[head];head++;
            for0(i,25){
                int v=next[now][i];
                if(!v)continue;
                int k=fail[now];
                while(!next[k][i])k=fail[k];
                fail[v]=next[k][i];
                q[tail++]=v;
            }
        }
        for(int i=tail-1;i>=0;i--)
            sum[fail[q[i]]]+=sum[q[i]];
    }
}acm;
int main(){
    //freopen("input.txt","r",stdin);
    //freopen("output.txt","w",stdout);
    n=read();
    for1(i,n)
        acm.insert(pos[i]);
    acm.build();
    for1(i,n)
        printf("%d\n",acm.sum[pos[i]]);
    return 0;
}