bzoj 2434 [Noi2011]阿狸的打字机(fail树模板)

版权声明:版权声明:本文为博主原创文章,未经博主允许不得转载,欢迎添加友链。 https://blog.csdn.net/zzk_233/article/details/82843620

先介绍一下fail树。

首先应该先学习ac自动机或者trie图

好了,现在大家都知道ac自动机和trie图是如何使用的了(就当大家知道了),那么什么是fail树呢,就是把fail指针反建,因为fail指

针一定只指向一个位置,所以它们连起来就是一个树的样子,而根据fail指针建立的过程,我们可以确定一个fail指针反建之后指向

的点一定以当前点作为后缀,并且它的子树都以它为后缀。而在trie树中,每个点都可以代表一个字符串的前缀。

那么举一个栗子,字符串abcac中点b的位置在trie树中可以代表前缀ab,那么反建之后,b点的位置的子树都是以ab为后缀的字符串

通过这个性质,我们就可以构建fail树之后,查找某个字符的位置(即代表一个字符串 )的子树中有多少个对他有贡献的点。这个

地方可以使用dfs序。

#include<cstdio>
#include<cmath>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
struct nn
{
	int ch[27];
	int fail,fa;
}t[100005];
struct node
{
	int to;
	int nxt;
}edge[100005];
struct nw
{
	int st,bz,pos;
}p[100005];
int head[100005],cnt=1;
char ss[100005];
int root,l,sz,n,tot,m,tt,t1=1;
int into[100005],out[100005],rt[100005],ans[100005],c[200005];
void init()
{
	memset(head,-1,sizeof(head));
}
void add(int from,int to)
{
	edge[cnt].to=to;
	edge[cnt].nxt=head[from];
	head[from]=cnt++;
}
void insert(int x,int val)
{
	if(!x)return;
	while(x<=l*2)
	{
		c[x]+=val;
		x+=(x&-x);
	}
}
int gett(int x)
{
	if(!x)return 0;
	int s=0;
	while(x)
	{
		s+=c[x];
		x-=(x&-x);
	}
	return s;
}
void get_fail()
{
	queue<int>M;
	int rt=0;
	for(int i=0;i<26;i++)
	{
		if(t[rt].ch[i])M.push(t[rt].ch[i]),add(rt,t[rt].ch[i]);
	}
	while(!M.empty())
	{
		rt=M.front();M.pop();
		for(int i=0;i<26;i++)
		{
			if(t[rt].ch[i])
			{
				t[t[rt].ch[i]].fail=t[t[rt].fail].ch[i];
				M.push(t[rt].ch[i]);
				add(t[t[rt].ch[i]].fail,t[rt].ch[i]);
			}else
			{
				t[rt].ch[i]=t[t[rt].fail].ch[i];
			}
		}
	}
}
void dfs(int u)
{
	into[u]=++tot;
	for(int i=head[u];i!=-1;i=edge[i].nxt)
	{
		int to=edge[i].to;
		dfs(to);
	}
	out[u]=tot;
}
int cmp(nw a,nw b)
{
	return a.bz<b.bz;
}
int main()
{
	init();
	scanf("%s",ss+1);
	l=strlen(ss+1);
	for(int i=1;i<=l;i++)
	{
		if(ss[i]=='B')
		{
			root=t[root].fa;
		}else if(ss[i]=='P')
		{
			rt[++n]=root;
		}else
		{
			int k=ss[i]-'a';
			if(!t[root].ch[k])t[root].ch[k]=++sz,t[sz].fa=root;
			root=t[root].ch[k];
		}
	}
	get_fail();
	dfs(0);
	scanf("%d",&m);
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&p[i].st,&p[i].bz);p[i].pos=i;
	}
	sort(p+1,p+1+m,cmp);root=0;
	for(int i=1;i<=l;i++)
	{
		if(ss[i]=='B')
		{
			insert(into[root],-1);
			root=t[root].fa;
		}else if(ss[i]=='P')
		{
			tt++;
			while(tt==p[t1].bz)
			{
				ans[p[t1].pos]=gett(out[rt[p[t1].st]])-gett(into[rt[p[t1].st]]-1);
				t1++;		
			}
		}else
		{
			int k=ss[i]-'a';
			root=t[root].ch[k];
			insert(into[root],1);
		}
	}
	for(int i=1;i<=m;i++)
	{
		printf("%d\n",ans[i]);
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/zzk_233/article/details/82843620