首页 > 代码库 > BZOJ 3172 Tjoi2013 单词 fail树

BZOJ 3172 Tjoi2013 单词 fail树

题目大意及后缀数组做法见 http://blog.csdn.net/popoqqq/article/details/41042473

原来正解是fail树……难怪后缀数组被卡成这样

首先我们将给出的n个串构建AC自动机

朴素的做法是对于每个串将这个串每个节点沿着fail指针扫一遍,将路径上的所有点的cnt++

但是这样做会TLE

我们不妨反向思考 fail指针反向后是一棵树 沿着fail指针扫一遍就是沿着树边向根扫一遍

只在插入时将每个串的每个节点cnt++ 那么每个串终点所在fail树的子树中cnt的总和就是这个字符串的出现次数

于是每个点记录所在fail树的子树中cnt的数量 可以线性时间内出解

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#define M 1001001
using namespace std;
struct Trie{
	Trie *son[26],*fail;
	int cnt;
}*root,mempool[M],*C=mempool;
int n;
Trie *pos[210];
char s[M];
void Insert(Trie* &p,char *s,Trie* &pos)
{
	if(!p) p=C++;
	p->cnt++;
	if(!*s)
	{
		pos=p;
		return ;
	}
	Insert(p->son[(*s)-'a'],s+1,pos);
}
void Build_Tree()
{
	static Trie *q[M];
	static int r,h;
	int i;
	for(i=0;i<26;i++)
		if(root->son[i])
			root->son[i]->fail=root,q[++r]=root->son[i];
	while(r!=h)
	{
		Trie *p=q[++h];
		for(i=0;i<26;i++)
			if(p->son[i])
			{
				Trie *temp=p->fail;
				while(temp!=root&&!temp->son[i])
					temp=temp->fail;
				if(temp->son[i]) temp=temp->son[i];
				p->son[i]->fail=temp;
				q[++r]=p->son[i];
			}
	}
	for(i=r;i;i--)
		q[i]->fail->cnt+=q[i]->cnt;
}
int main()
{
	int i;
	cin>>n;
	for(i=1;i<=n;i++)
		scanf("%s",s),Insert(root,s,pos[i]);
	Build_Tree();
	for(i=1;i<=n;i++)
		printf("%d\n",pos[i]->cnt);
	return 0;
}


BZOJ 3172 Tjoi2013 单词 fail树