标算是把路径修改单点求值 改为 单点修改子树查询,这样只要再在AC自动机上跑一边,然后把询问相同的用vector挂个链,碰到单词结尾查询一下子树和即可,可以用树状数组实现;
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
const int N=200050;
int trie[N][26],fa[N],fail[N],pos[N],len,tt,cnt;
int head[N*26],nxt[N*26],to[N*26],tot,flag,dfn[N],ed[N],T,c[N],ans[N];
char a[N];
void lnk(int x,int y){
to[++tot]=y,nxt[tot]=head[x],head[x]=tot;
}
void insert(){
int x=0;
for(int i=1;i<=len;i++){
if(a[i]==‘P‘) pos[++tt]=x;
else if(a[i]==‘B‘) x=fa[x];
else{
if(!trie[x][a[i]-‘a‘]) trie[x][a[i]-‘a‘]=++cnt;
fa[cnt]=x,x=trie[x][a[i]-‘a‘];
}
}
}
void build(){
queue<int> q; fail[0]=0;
for(int i=0;i<26;i++){
if(trie[0][i]) q.push(trie[0][i]);
}
while(!q.empty()){
int u=q.front();q.pop();
for(int i=0;i<26;i++){
int x=trie[u][i],j=fail[u];
if(!x) continue;
q.push(x);
while(j&&!trie[j][i]) j=fail[j];
if(trie[j][i]) fail[x]=trie[j][i];
}
}
for(int i=1;i<=cnt;i++) lnk(fail[i],i);
}
void dfs(int x){
dfn[x]=++T;
for(int i=head[x];i;i=nxt[i]) dfs(to[i]);
ed[x]=T;
}
void update(int x,int val){
for(int i=x;i<=T;i+=i&-i) c[i]+=val;
}
int query(int x){
int sum=0;
for(int i=x;i;i-=i&-i) sum+=c[i];
return sum;
}
struct data{
int x,id;
};
vector<data> p[N];
void solve(){
int m;cin>>m;
for(int i=1;i<=m;i++){
int x,y;scanf("%d%d",&x,&y);
p[pos[y]].push_back((data){x,i});
}
int x=0;
for(int i=1;i<=len;i++){
if(a[i]==‘P‘){
for(int j=0;j<p[x].size();j++){
data y=p[x][j];
ans[y.id]=query(ed[pos[y.x]])-query(dfn[pos[y.x]]-1);
}
}
else if(a[i]==‘B‘) update(dfn[x],-1),x=fa[x];
else x=trie[x][a[i]-‘a‘],update(dfn[x],1);
}
for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}
int main(){
scanf("%s",a+1);len=strlen(a+1);
insert();build();dfs(0);solve();
}