BZOJ 3172 [Tjoi2013]单词 AC自动机 模板题

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

题目传送门

开心~又在bzoj找到一题水题。。

注意一下题目题意有点点不清晰。。。

其实总的串是N个单词全部连接起来，而且每两个之间有一个空格（也可以用其它字符代替）

还有就是是N个单词全部连起来的长度<=10^6……

题目比较良心的地方是内存开了512M，这样子的话可以直接上trie[][]了。

总之是一道水水的模板题吧。。

感觉这题加快速读入就是智障

#include<bits/stdc++.h>
using namespace std;
const int
	N=1000205,
	NUM=205;
int n,cnt,LenAll;
int pp[N];
int Ans[NUM],fail[N],num[N],Q[N];
int trie[N][27];
char All[N],tmp[N];
int find(int x,int t){
	if (trie[x][t]) return trie[x][t];
	if (!x) return 0;
	return find(fail[x],t);
}
void insert(int ii){
	int len=strlen(tmp),now=0;
	for (int i=0;i<len;i++){
		int j=tmp[i]-'a';
		if (trie[now][j]) now=trie[now][j];
			else now=trie[now][j]=++cnt;
	}
	if (num[now]) pp[ii]=num[now];
		else num[now]=ii;
}
void AC(){
	int head=0,tail=1;
	Q[0]=0;
	while (head!=tail){
		int now=Q[head++];
		for (int i=0;i<27;i++){
			if (!trie[now][i]) continue;
			int t=find(fail[now],i);
			if (!now) t=0;
			fail[trie[now][i]]=t;
			Q[tail++]=trie[now][i];
		}
	}
}
void solve(){
	int now=0;
	for (int i=0;i<LenAll;i++){
		int j=All[i]-'a';
		now=find(now,j);
		for (int j=now;j;j=fail[j])
			if (num[j]) Ans[num[j]]++;
	}
}
int main(){
    scanf("%d",&n);
	LenAll=cnt=0;
	for (int i=1;i<=n;i++){
		scanf("%s",tmp);
		int len=strlen(tmp);
		for (int j=0;j<len;j++) All[LenAll++]=tmp[j];
		All[LenAll++]='{';
		insert(i);
	}
	AC();
	solve();
	for (int i=1;i<=n;i++)
        if (pp[i]) Ans[i]=Ans[pp[i]];
	for (int i=1;i<=n;i++)
		printf("%d\n",Ans[i]);
	return 0;
}