TJOI2013单词

链接：https://www.luogu.org/problemnew/show/P3966

描述

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

输入

第一个一个整数 N ，表示有多少个单词，接下来 N 行每行一个单词。

输出

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

样例输入

3
a
aa
aaa

样例输出

6
3
1

提示

对于全部数据，1≤N≤200 ，所有单词长度的和不超过 106，保证每个单词由小写字母组成。

#include<cstdio>//网上找的
#include<cstring>
#include<iostream>
#include<algorithm>
#define N 1100005
using namespace std;
int n,a[N],h[N],cnt,last,ch[N][26],sz[N],fail[N];
char s[N];
struct ac{
    void ins(int x){
        scanf("%s",s+1);int now=0,len=strlen(s+1);
        for(int i=1;i<=len;i++){
            int u=s[i]-'a';
            if(!ch[now][u]) ch[now][u]=++cnt;
            now=ch[now][u];
            sz[now]++;//
        }
        a[x]=now;
    }
    void build(){
        int i,head=0,tail=0;
        for(i=0;i<26;i++) if(ch[0][i]) h[++tail]=ch[0][i];//h数组按深度中记录了所有的点 
        while(head<tail){
            int x=h[++head],y;
                for(i=0;i<26;i++) if(y=ch[x][i]){//先令y=ch[x][i],在判断y是否为0 
                h[++tail]=y; 
                fail[y]=ch[fail[x]][i];
            }
            else ch[x][i]=ch[fail[x]][i];
        }
    }
    void solve(){
        for(int i=cnt;i>=0;i--) sz[fail[h[i]]]+=sz[h[i]];//h数组是广搜进行赋值，最终的由cnt（点的总数）到1所指向的点的深度依次减小，保证sz数组的累加不出差错 
        for(int i=1;i<=n;i++) printf("%d\n",sz[a[i]]);
    }
}ac;
int main(){
    //freopen("word.in","r",stdin);
    //freopen("word.out","w",stdout);
    scanf("%d",&n);
    for(int i=1;i<=n;i++) ac.ins(i);
    ac.build();ac.solve();
    return 0;
}

#include<cstdio>//200个单词，长度1e6，自己编的但过不了 
#include<bits/stdc++.h>
#include<cstring>
using namespace std;
const int N=11e5+5;
/*struct fjy{
	int num;char w[N];
}stc[205]; */
int trie[N][30],fail[N],val[N],ans[205],que[N],que2[N],q=0;
int n,cnt=0;
char w[N];
void insert(char *x,int num){
	int u=0,len=strlen(x);
	for(int i=0;i<len;i++){
		int c=x[i]-'a';
		if(!trie[u][c]) trie[u][c]=++cnt;
		u=trie[u][c];ans[u]++;
	}
	//ans[u]++不能放在外面 
	que2[num]=u;
}
void getfail(){
	int q1=1,q2=0;
	for(int i=0;i<26;i++)
		if(trie[0][i]) que[++q2]=trie[0][i];
	for(;q1<=q2;q1++){
		int u=que[q1];
		for(int i=0;i<26;i++){
			if(trie[u][i]) que[++q2]=trie[u][i],fail[trie[u][i]]=trie[fail[u]][i];
			else trie[u][i]=trie[fail[u]][i];
		}
	}
}
/*void find(char *s){
	int u=0,len=strlen(s);
	for(int i=0;i<len;i++){
		int c=s[i]-'a';
		u=trie[u][c];
		for(int j=u;j;j=fail[j]) ans[val[j]]++;
	}
}*/

int main(){
	scanf("%d",&n);
	for(int i=1;i<=n;i++){
		scanf("%s",w);
		insert(w,i);
	}
	getfail();
	for(int i=cnt;i>0;i--){
		ans[fail[que[i]]]+=ans[que[i]];
	}
	for(int i=1;i<=n;i++) printf("%d\n",ans[que2[i]]);
	return 0;
}