题意
给定n个模式串,有m个询问,每次询问第X个模式串在第Y个模中出现了多少次
解题思路
以fail树相反的方向建一棵树T,问题转化为X的子树中有多少个y的终止节点。跑出T的dfs序,X的子树就可以表示为一段区间,然后对trie跑一遍dfs,每进入一个点就把dfs序列上对应的位置+1,离开一个点就把dfs序列上对应的位置-1,遍历到y的终止节点就计算所有Y=y的询问的结果,计算方法就是区间求和,用树状数组或线段树可以轻易地维护。
AC代码
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=2e5+5;
char s[maxn];
struct Query{
int x,y,id,ans;
bool operator<(const Query& b)const{return y<b.y;}
}q[maxn];
int ql[maxn],qr[maxn],pos[maxn];
//BIT
int c[maxn];
inline int lowbit(int x){return x&(-x);}
void add(int x,int d){for(;x<maxn-5;x+=lowbit(x))c[x]+=d;}
int getsum(int x){int res=0;for(;x;x-=lowbit(x))res+=c[x];return res;}
int getsum(int l,int r){return getsum(r)-getsum(l-1);}
//AC_Automaton
//root=0,range[1,tot]
const static int SIZE=26;
int tr[maxn][SIZE],fail[maxn],tot;
int fa[maxn],val[maxn],ch[maxn][SIZE];
void insert(){
int p=0,cnt=0;
for(int i=0;s[i];i++){
if(s[i]=='P'){pos[++cnt]=p;val[p]=cnt;}
else if(s[i]=='B')p=fa[p];
else{
if(!tr[p][s[i]-'a'])tr[p][s[i]-'a']=++tot;
fa[tr[p][s[i]-'a']]=p;
p=tr[p][s[i]-'a'];
}
}
for(int i=0;i<=tot;i++)for(int j=0;j<26;j++)ch[i][j]=tr[i][j];
}
void getfail(){
queue<int>q;
for(int i=0;i<SIZE;i++)if(tr[0][i])fail[tr[0][i]]=0,q.push(tr[0][i]);
while(!q.empty()){
int p=q.front();q.pop();
for(int i=0;i<SIZE;i++){
if(tr[p][i]){
fail[tr[p][i]]=tr[fail[p]][i],q.push(tr[p][i]);
}
else tr[p][i]=tr[fail[p]][i];
}
}
}
//inv fail tree
struct Edge{
int v,nxt;
}e[maxn];
int fi[maxn],se[maxn],sz;
int head[maxn],num;
void init_graph(){
memset(head,0,sizeof(head));
num=1;
}
void addedge(int u,int v){e[num].v=v;e[num].nxt=head[u];head[u]=num++;}
void dfs_fail(int p){
fi[p]=++sz;
for(int i=head[p];i;i=e[i].nxt)dfs_fail(e[i].v);
se[p]=sz;
}
//get ans
int Ans[maxn];
void dfs_trie(int p){
add(fi[p],1);
if(val[p]){
for(int i=ql[val[p]];i<=qr[val[p]];i++)
q[i].ans=getsum(fi[pos[q[i].x]],se[pos[q[i].x]]);
}
for(int i=0;i<26;i++)if(ch[p][i])dfs_trie(ch[p][i]);
add(fi[p],-1);
}
int main()
{
//#ifndef ONLINE_JUDGE
// freopen("in.txt","r",stdin);
//#endif
scanf("%s",s);
insert();
getfail();
init_graph();
for(int i=1;i<=tot;i++)addedge(fail[i],i);
sz=0;dfs_fail(0);//建inv fail tree,跑出dfs序
int n;
scanf("%d",&n);
for(int i=1;i<=n;i++){
scanf("%d %d",&q[i].x,&q[i].y);
q[i].id=i;q[i].ans=0;
}
sort(q+1,q+1+n);
for(int i=1,t=1;i<=n;i++){
ql[q[i].y]=i;
while(q[t+1].y==q[i].y)t++;
qr[q[i].y]=t;i=t;
}
dfs_trie(0);//遍历trie,跑出答案
for(int i=1;i<=n;i++)Ans[q[i].id]=q[i].ans;
for(int i=1;i<=n;i++)printf("%d\n",Ans[i]);
return 0;
}