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+10;
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[26],f,s;
}t[N<<1];
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<<1];
inline int lowbit(int x){return x&(-x);}
int query(int x)
{
    int ans=0;
    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=0;
    for(int i=0;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=0;i<26;++i)
    {
        if(t[0].v[i])
        q.push(t[0].v[i]),t[t[0].v[i]].f=0;
    }
    while(!q.empty())
    {
        int x=q.front();q.pop();
        for(int i=0;i<26;++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=1;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=1;i<=m;++i)
    {
        scanf("%d%d",&x,&y);add2(y,x);
    }int now=0;
    for(int i=0;i<n;++i)
    {
        if(s[i]=='B')
        {
            update(l[now],-1);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]-1);
            }
        }
        else
        {
            now=t[now].v[s[i]-'a'];
            update(l[now],1);
        }
    }
    for(int i=1;i<=m;++i)
    printf("%d
",ans[i]);
    return;
}
int main()
{
    scanf("%s",s);
    n=strlen(s);
    build();
    getfail();
    maketree();
    dfs(0);
    work();
    return 0;
}