【BZOJ 2434】[Noi2011]阿狸的打字机 Ac自动机+树状数组+dfs序

很简单的想法是，建造Ac自动机以后，如果一个单词节点的fail指针指向另外一个字符串的路径上，那么它对两者之间会造成贡献，基于这样的想法，我们需要做的就是如何快速的求这个贡献。考虑反向链接fail指针每个节点只有一个出边所以将会是一个树，根据刚才的性质，根节点会对他的所有子树中的节点造成答案的贡献，然后维护dfs序，就能快速知道他的子树区间。最后统计答案的时候离线处理。对于一个串，将Trie数上的路径上的点对应的dfs序++（用树状数组维护），然后扫一遍问题，此时他作为y，那么就查找所有问到它作为y的问题，查询x的子树的区间和就是答案（根据刚才说的一个根节点一定是被包含在所有子树节点中）至此问题解决

#include<cstdio>
#include<cstring>
#include<iostream>
#include<vector>
#define maxn 200020
using namespace std;
int ch[maxn][26],fail[maxn],cnt,cur[maxn*2],q[maxn*2],head[maxn],tot,size,val[maxn];
int in[maxn*2],out[maxn*2],n,nu[maxn*2],dep[maxn*2],c[maxn*4],len,ff[maxn*2];
bool vis[maxn*2][26];
char ss[maxn*2],cc[maxn*2];
struct node{int a,b,ans;}nod[maxn*2];
struct edge{int v,next;}e[maxn*2];
vector<int>g[maxn];
void adde(int a,int b){e[tot].v=b,e[tot].next=head[a];head[a]=tot++;}
void update(int x,int add){while(x<=cnt){c[x]+=add;x+=x&(-x);}}
int query(int x){
	int ans=0;
	while(x>0){ans+=c[x];x-=x&(-x);}
	return ans;
}
void insert(){
	int rt=0,id=0;
	for(int y,j=0;j<len;j++){
		if(ss[j]=='P')val[rt]=++id,nu[id]=rt;
		else if(ss[j]=='B')rt=ff[rt];
		else{
			y=ss[j]-'a';
			if(!ch[rt][y])ch[rt][y]=++size;
			ff[ch[rt][y]]=rt;
			rt=ch[rt][y];
		}
	}
}
void build_ac(){
	int l=0,r=1;q[0]=0;
	while(l<r){
		int u=q[l++];
		for(int i=0;i<26;i++){
			if(ch[u][i])fail[ch[u][i]]=u==0?0:ch[fail[u]][i],q[r++]=ch[u][i];
			else ch[u][i]=u==0?0:ch[fail[u]][i],vis[u][i]=true;
		}
	}
	for(int i=1;i<=size;i++){
		adde(fail[i],i);
	}
}
void dfs(int u){
	in[u]=++cnt;
	for(int v,i=head[u];i!=-1;i=e[i].next){
		v=e[i].v;
		dfs(v);
	}
	out[u]=cnt;
}

void solve(){
	int rt=0;
	for(int j=0;j<len;j++){
		if(ss[j]=='P'){
			for(int i=0;i<g[val[rt]].size();i++){
				int y=nu[nod[g[val[rt]][i]].a];//对应的那一个串的节点标号 
				int ans=query(out[y])-query(in[y]-1);
				nod[g[val[rt]][i]].ans=ans;
			}
		}
		else if(ss[j]=='B')update(in[rt],-1),rt=ff[rt];
		else rt=ch[rt][ss[j]-'a'],update(in[rt],1);
	}
}
int main(){
	memset(head,-1,sizeof(head));
	scanf("%s",ss);len=strlen(ss);
	insert();
	build_ac();
	dfs(0);
	scanf("%d",&n);int ccnt=0;
	for(int i=1;i<=n;i++){
		ccnt++;
		scanf("%d%d",&nod[ccnt].a,&nod[ccnt].b);
		g[nod[ccnt].b].push_back(ccnt);
	}
	solve();
	for(int i=1;i<=ccnt;i++)printf("%d\n",nod[i].ans);
	return 0;
}