【BZOJ 2434】 [Noi2011]阿狸的打字机 fail树+树状数组

#include <cstdio>
#include <cstring>
#include <iostream>
#include <vector>
const int N=100010;
char s
;
struct Trie{
int ch[26],fail,fa,deep;
}node
;
std::vector<int> mem
;
int sz;
struct V{
int to,next;
}c
;
struct VQ{
int to,next,id;
}ques
;
int head
,t,num,ques_head
,ques_t;
int belong
,q
,l
,r
,Time;
int T
,ans
,m,n;
inline int Q(int pos){
int ret=0;
for(;pos>0;pos-=pos&(-pos))
ret+=T[pos];
return ret;
}
inline void U(int pos,int key){
for(;pos<=n;pos+=pos&(-pos))
T[pos]+=key;
}
inline void add(int x,int y){
c[++t].to=y,c[t].next=head[x],head[x]=t;
}
inline void ques_add(int x,int y,int z){
ques[++ques_t].to=y,ques[ques_t].next=ques_head[x],ques_head[x]=ques_t,ques[ques_t].id=z;
}
void dfs(int x){
l[x]=++Time;
for(int i=head[x];i;i=c[i].next)
dfs(c[i].to);
r[x]=Time;
}
void dfs_(int x){
U(l[x],1);
for(int i=0;i<mem[x].size();i++)
for(int j=ques_head[mem[x][i]];j;j=ques[j].next)
ans[ques[j].id]=Q(r[belong[ques[j].to]])-Q(l[belong[ques[j].to]]-1);
for(int i=0;i<26;i++)
if(node[node[x].ch[i]].deep>node[x].deep)
dfs_(node[x].ch[i]);
U(l[x],-1);
}
int main(){
scanf("%s",s);
int now=0;
for(int i=0;s[i];i++){
if(s[i]=='P'){
belong[++num]=now;
mem[now].push_back(num);
continue;
}
if(s[i]=='B'){
now=node[now].fa;
continue;
}
if(!node[now].ch[s[i]-'a'])node[now].ch[s[i]-'a']=++sz,node[sz].fa=now,node[sz].deep=node[now].deep+1;
now=node[now].ch[s[i]-'a'];
}
n=sz+1;
q[0]=0;
for(int i=0,j=0;i<=j;i++)
for(int l=0;l<26;l++)
if(node[q[i]].ch[l]){
q[++j]=node[q[i]].ch[l];
node[q[j]].fail=q[i]?node[node[q[i]].fail].ch[l]:0;
add(node[q[j]].fail,q[j]);
}else
node[q[i]].ch[l]=q[i]?node[node[q[i]].fail].ch[l]:0;
dfs(0);
scanf("%d",&m);
for(int i=1,x,y;i<=m;i++)
scanf("%d%d",&x,&y),ques_add(y,x,i);
dfs_(0);
for(int i=1;i<=m;i++)
printf("%d\n",ans[i]);
return 0;
}