2434: [Noi2011]阿狸的打字机

题目链接

题解：

这个人的题解好：http://blog.csdn.net/huzecong/article/details/7769988

首先构建ac自动机的fail树

得出dfs序，得出每个结点进出时间l[x]，r[x]，

然后用树状数组维护

#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=200005;
int n,m;
char s[N];
int sa[N][100],tot=1,fail[N],pos[N],id1=0,fa[N];
struct node{
	int x,y,next;
}tr[N];int len=0,first[N];
struct node1{
	int x,y,next;
}de[N];int len1=0,next[N];
void insert(int x,int y)
{
	len++;
	tr[len].x=x;
	tr[len].y=y;
	tr[len].next=first[x];
	first[x]=len;
}
void ins()
{
	int now=1,id=0;
		for(int i=0;i<n;i++)
			if(s[i]=='P')pos[++id]=now;
			else if(s[i]=='B')now=fa[now];
			else 
			{
				if(!sa[now][s[i]-'a'])
				{
					sa[now][s[i]-'a']=++tot;
					fa[tot]=now;
				}
				now=sa[now][s[i]-'a'];
			}
}
int q[N];
void makefail()
{
    int head=0,tail=1;
		fail[1]=0;q[0]=1;
		while(head!=tail)
		{
			int now=q[head];head++;
			for(int i=0;i<26;i++)
				if(sa[now][i])
				{
					int v=sa[now][i];
					int k=fail[now];
					while(!sa[k][i])k=fail[k];
					fail[v]=sa[k][i];
					q[tail++]=v;
				}
		}
}
int t[N],ans[N];int T=0;
void add(int x,int val)
{
	for(int i=x;i<=T;i+=i&-i) t[i]+=val;
}
int query(int x)
{
	int sum=0;
	for(int i=x;i;i-=i&-i)sum+=t[i];
	return sum;
}
int l[N],r[N];
void solve(){
		int now=1,id=0;
		add(l[1],1);
		for(int i=0;i<n;i++)
			if(s[i]=='P')
			{
				id++;
				for(int x=next[id];x!=-1;x=de[x].next)
				{
					int t=pos[de[x].y];
					ans[x]=query(r[t])-query(l[t]-1);
				}
			}
			else if(s[i]=='B')add(l[now],-1),now=fa[now];
			else now=sa[now][s[i]-'a'],add(l[now],1);
	}
void dfs(int x)
{
	l[x]=++T;
	for(int i=first[x];i!=-1;i=tr[i].next)
	dfs(tr[i].y);
	r[x]=++T;
}
int main()
{
	scanf("%s",s);
	n=strlen(s);
	memset(first,-1,sizeof(first));
	memset(next,-1,sizeof(next));
	for(int i=0;i<26;i++)
	sa[0][i]=1;
	ins();//	
	makefail();
	for(int i=1;i<=tot;i++)
	insert(fail[i],i);
	scanf("%d",&m);
	int x,y;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d",&x,&y);
		len1++;de[len1].x=x;de[len1].y=x;
		de[len1].next=next[y];next[y]=len1;
	}
	memset(l,0,sizeof(l));
	memset(r,0,sizeof(r));
	dfs(0);
	memset(t,0,sizeof(t));
	solve();
	for(int i=1;i<=m;i++)
	printf("%d\n",ans[i]);
}