[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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值