BZOJ 3172 【Tjoi2013】单词

3172: [Tjoi2013]单词

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

【题目分析】

    思路十分巧妙，每次先匹配再插入会超时的，记录下标之后，从较深的结点向上递推累加，画出图来就很容易理解了。刚开始数组开小了，TLE(好奇竟然不是RE)。

【代码】

#include <cstdio>
#include <cstring>
#include <queue>
using namespace std;
int cnt=0,list[1000005],fail[1000005],trie[1000005][26],pos[2001],v[1000005],tot=1,n;
char s[1000001];
inline void insert(int num)
{
	scanf("%s",s);
	int now=1,l=strlen(s)-1;
	for (int i=0;i<=l;++i)
	{
		if (!trie[now][s[i]-'a']) trie[now][s[i]-'a']=++tot;
		now=trie[now][s[i]-'a'];
		v[now]++;//经过该节点的数目 
	}
	pos[num]=now;//记录每一个字符串的最终位置 
}
inline void bfs()
{
	list[++cnt]=1;
	queue<int>q;
	q.push(1);
	while(!q.empty()){
		int x=q.front(); q.pop();
		list[++cnt]=x;//记录bfx序 
		for (int i=0;i<=25;++i){
			int j=fail[x];
			while (j&&!trie[j][i]) j=fail[j];
			if (trie[x][i]) {fail[trie[x][i]]=j?trie[j][i]:1;q.push(trie[x][i]);}
			else trie[x][i]=j?trie[j][i]:1;
		}
	}
}
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;++i) insert(i);
	bfs();
	for (int i=cnt;i>=1;i--){
		v[fail[list[i]]]+=v[list[i]];
	}//逆bfs序递推 
	for (int i=1;i<=n;++i){
		printf("%d\n",v[pos[i]]);
	}
}