BZOJ2434 NOI2011阿狸的打字机

询问x这个串在y中出现的次数。

fail数组有一个性质就是一旦a的fail指向b那么b所代表的字串一定是a的后缀。

所以我们看fail树（即按fail反向建树）中x的子树有多少y的结点即可。

这个操作可以使用树状数组维护dfs序操作。

By：大奕哥

 #include<bits/stdc++.h>
 using namespace std;
 const int N=1e5+;
 int head[N],dead[N],cnt,ent,dnt,num,n,m,idx,fa[N],pos[N],l[N],r[N];
 char s[N];
 queue<int>q;
 struct poin{
     int v[],f,s;
 }t[N<<];
 struct node{
     int to,nex;
 }e[N];
 struct edge{
     int to,nex;
 }d[N];
 void add(int x,int y)
 {
     e[++ent].to=y;e[ent].nex=head[x];head[x]=ent;
 }
 void add2(int x,int y)
 {
     d[++dnt].to=y;d[dnt].nex=dead[x];dead[x]=dnt;
 }
 int tr[N<<];
 inline int lowbit(int x){return x&(-x);}
 int query(int x)
 {
     int ans=;
     for(;x;x-=lowbit(x))ans+=tr[x];
     return ans;
 }
 void update(int x,int w)
 {
     for(;x<=idx;x+=lowbit(x))tr[x]+=w;
 }
 void build()
 {
     int now=;
     for(int i=;i<n;++i)
     {
         if(s[i]=='P')pos[++num]=now;
         else if(s[i]=='B')now=fa[now];
         else{
             if(!t[now].v[s[i]-'a'])
                 t[now].v[s[i]-'a']=++cnt;
             fa[t[now].v[s[i]-'a']]=now;
             now=t[now].v[s[i]-'a'];
         }
     }
     return;
 }
 void getfail()
 {
     for(int i=;i<;++i)
     {
         if(t[].v[i])
         q.push(t[].v[i]),t[t[].v[i]].f=;
     }
     while(!q.empty())
     {
         int x=q.front();q.pop();
         for(int i=;i<;++i)
         {
             if(t[x].v[i])
             {
                 t[t[x].v[i]].f=t[t[x].f].v[i];
                 q.push(t[x].v[i]);
             }
             else{
                 t[x].v[i]=t[t[x].f].v[i];
             }
         }
     }
     return;
 }
 void maketree()
 {
     for(int i=;i<=cnt;++i)
     {
         int x=t[i].f;int y=i;
         add(x,y);
     }
     return;
 }
 void dfs(int x)
 {
     l[x]=++idx;
     for(int i=head[x];i;i=e[i].nex)
     {
         int y=e[i].to;
         dfs(y);
     }
     r[x]=++idx;
     return;
 }
 int ans[N];
 void work()
 {
     scanf("%d",&m);int id;int x,y;
     for(int i=;i<=m;++i)
     {
         scanf("%d%d",&x,&y);add2(y,x);
     }int now=;
     for(int i=;i<n;++i)
     {
         if(s[i]=='B')
         {
             update(l[now],-);now=fa[now];
         }
         else if(s[i]=='P')
         {
             ++id;
             for(int j=dead[id];j;j=d[j].nex)
             {
                 int y=pos[d[j].to];
                 ans[j]=query(r[y])-query(l[y]-);
             }
         }
         else
         {
             now=t[now].v[s[i]-'a'];
             update(l[now],);
         }
     }
     for(int i=;i<=m;++i)
     printf("%d\n",ans[i]);
     return;
 }
 int main()
 {
     scanf("%s",s);
     n=strlen(s);
     build();
     getfail();
     maketree();
     dfs();
     work();
     return ;
 }