[NOI2011] 阿狸的打字机

• 把询问全部放到树上，离线dfs处理

点击查看代码

#include <bits/stdc++.h>
using namespace std;
string s;
int t[100005][26],tot,cnt,r[100005],fa[100005],fail[100005],dfn[100005],w[100005],sum,ans[100005];
bool b[100005];
vector<int>a[100005];
vector<int>o[100005];
vector<int>g[100005];
queue<int>q;
struct u1
{
	int id,x,y;
}u[100005];
int read1()
{
	char cc=getchar();
	while(!(cc>=48&&cc<=57))
	{
		if(cc=='-')
		{
			break;
		}
		cc=getchar();
	}
	bool f=false;
	int s=0;
	if(cc=='-')
	{
		f=true;
	}
	else
	{
		s=cc-48;
	}
	while(1)
	{
		cc=getchar();
		if(cc>=48&&cc<=57)
		{
			s=s*10+cc-48;
		}
		else
		{
			break;
		}
	}
	if(f==true)
	{
		s=-s;
	}
	return s;
}
void build()
{
	int cur=0;
	for(int i=0;i<s.size();i++)
	{
		if(s[i]=='B')
		{
			cur=fa[cur];
		}
		else if(s[i]=='P')
		{
			cnt++;
			r[cnt]=cur;
			b[cur]=true;
		}
		else
		{
			if(t[cur][s[i]-'a']==0)
			{
				tot++;
				t[cur][s[i]-'a']=tot;
				fa[tot]=cur;
				o[cur].push_back(tot);
			}
			cur=t[cur][s[i]-'a'];
		}
	}
	for(int i=0;i<26;i++)
	{
		if(t[0][i]!=0)
		{
			q.push(t[0][i]);
		}
	}
	while(!q.empty())
	{
		int n1=q.front();
		q.pop();
		for(int i=0;i<26;i++)
		{
			if(t[n1][i]!=0)
			{
				fail[t[n1][i]]=t[fail[n1]][i];
				q.push(t[n1][i]);
			}
			else
			{
				t[n1][i]=t[fail[n1]][i];
			}
		}
	}
	for(int i=1;i<=tot;i++)
	{
		a[fail[i]].push_back(i);
	}
}
void dfs1(int n1)
{
	sum++;
	dfn[n1]=sum;
	w[n1]=1;
	for(int i=0;i<a[n1].size();i++)
	{
		dfs1(a[n1][i]);
		w[n1]+=w[a[n1][i]];
	}
}
int c[100005];
int lowbit(int n)
{
	return n&(-n);
}
void add(int n1,int va)
{
	while(n1<=sum)
	{
		c[n1]+=va;
		n1+=lowbit(n1);
	}
}
int ask(int n1)
{
	int s=0;
	while(n1>0)
	{
		s+=c[n1];
		n1-=lowbit(n1);
	}
	return s;
}
void dfs2(int n1)
{
	add(dfn[n1],1);
	for(int i=0;i<g[n1].size();i++)
	{
		int x=u[g[n1][i]].x;
		ans[g[n1][i]]=ask(dfn[x]+w[x]-1)-ask(dfn[x]-1);
	}
	for(int i=0;i<o[n1].size();i++)
	{
		dfs2(o[n1][i]);
	}
	add(dfn[n1],-1);
}
int main()
{
	cin>>s;
	build();
	dfs1(0);
	int m;
	cin>>m;
	for(int i=1;i<=m;i++)
	{
		u[i].id=i;
		u[i].x=r[read1()];
		u[i].y=r[read1()];
		g[u[i].y].push_back(u[i].id);
	}
	dfs2(0);
	for(int i=1;i<=m;i++)
	{
		printf("%d\n",ans[i]);
	}
	return 0;
}