[FROM LUOGU][NOI2011]阿狸的打字機

傳送門

SOL
首先構建AC自動機，然後建一棵Fail樹
Fail樹就是把Fail指針反轉後重新建的樹，這樣查詢(x,y)，y會出現在x的子樹中
對於詢問（x，y），即統計在x的路徑上的節點中有多少個在y的子樹中
一邊模擬打字過程，一邊用樹狀數組維護每個節點在其子樹中出現次數

代碼：

#include<bits/stdc++.h>
#define re register
using namespace std;
const int N=2e5+5,SIZ=26;
char t[N];int fail[N],rev[N],fa[N],ans[N],ch[N][SIZ],tot,num,stk[N];
typedef pair<int,int> T;vector<T>ve[N];
int m,in[N],out[N],dfn,cnt,first[N],c[N];
struct edge{int v,nxt;}e[N<<1];
inline void add(int u,int v){e[++cnt].v=v;e[cnt].nxt=first[u];first[u]=cnt;}
inline void insert(){
	int re len=strlen(t),u=0;
	for(int re i=0;i<len;i++){
		if(t[i]=='B')u=fa[u];
		else if(t[i]=='P')stk[++num]=u;
			 else{
			 	 int re c=t[i]-'a';
				 if(!ch[u][c])ch[u][c]=++tot,fa[ch[u][c]]=u;
				 u=ch[u][c];			
			 }
	}
}
inline void getfail(){
	queue<int>q;
	for(int re i=0;i<SIZ;i++)if(ch[0][i])q.push(ch[0][i]);
	while(!q.empty()){
		int re r=q.front();q.pop();
		for(int re i=0;i<SIZ;i++){
			int re u=ch[r][i];
			if(!u){ch[r][i]=ch[fail[r]][i];continue;}
			fail[u]=ch[fail[r]][i],q.push(u);			
		}
	}
	for(int re i=1;i<=tot;i++)add(i,fail[i]),add(fail[i],i);
}	
inline void pre(int u,int f){
	in[u]=++dfn;
	for(int re i=first[u];i;i=e[i].nxt){
		int re v=e[i].v;
		if(v==f)continue;
		pre(v,u);
	}out[u]=dfn;
}
inline int lb(int x){return x&-x;}
inline void edit(int x,int v){for(int re i=x;i<=dfn;i+=lb(i))c[i]+=v;}
inline int ask(int x){int ret=0;for(int re i=x;i;i-=lb(i))ret+=c[i];return ret;}
signed main(){
	scanf("%s",t);insert();getfail();pre(0,0);scanf("%d",&m);
	for(int re i=1;i<=m;i++){
		int re x,y;
		scanf("%d%d",&x,&y);
		ve[y].push_back(make_pair(x,i));
	}int re len=strlen(t),u=0,pos=0;
	for(int re i=0;i<len;i++){
		if(t[i]=='B')edit(in[u],-1),u=fa[u];
		else if(t[i]=='P'){
				++pos;
				int re siz=ve[pos].size();
				for(int re j=0;j<siz;j++){
					int re v=stk[ve[pos][j].first],k=ve[pos][j].second;
					ans[k]=ask(out[v])-ask(in[v]-1);
				}
			 }
			 else u=ch[u][t[i]-'a'],edit(in[u],1);
	}
	for(int re i=1;i<=m;i++)printf("%d\n",ans[i]);
	return 0;
}