洛谷P2414 [NOI2011]阿狸的打字机（AC自动机）

传送门

 

考虑一下，如果串B在串A中出现过，那么A的fail指针必定直接或间接指向B

那么我们可以把fail树建起来，那么就变成B代表的节点的子树里有多少节点属于A

然后这就是一个序列统计问题，直接用dfs序+树状数组可以维护

具体的操作就是，先把每一个点有关的询问给存起来，然后等到在trie树上一遍dfs，当dfs到这个串的结束节点说明所有有关这个串的节点都被遍历到（可以在遍历的时候顺便用树状数组维护序列和），然后对每一个询问查一下子树和即可

//minamoto
#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<algorithm>
using namespace std;
const int N=100005;
int tot=1;
struct node{
    int num,type;
    node(){}
    node(int num,int type):num(num),type(type){}
};
queue<node> q;
int ch[N][26],End[N],word,fa[N],fail[N],dis[N];
inline int insert(int p,int c){
    if(ch[p][c]) return ch[p][c];
    ch[p][c]=++tot,fa[tot]=p;return tot;
}
inline int back(int p){return fa[p];}
inline void ed(int p){End[p]=++word,dis[word]=p;}
void build(){
    for(int i=0;i<26;++i)
    if(ch[1][i])
    q.push(node(ch[1][i],i)),fail[ch[1][i]]=1;
    while(!q.empty()){
        node now=q.front();q.pop();
        if(fail[now.num]!=1){
            int trail=fail[fa[now.num]];
            while(true){
                if(ch[trail][now.type]){trail=ch[trail][now.type];break;}
                if(trail==1) break;trail=fail[trail];
            }
            fail[now.num]=trail;
        }
        for(int i=0;i<26;++i)
        if(ch[now.num][i])
        q.push(node(ch[now.num][i],i));
    }
}
int ver[N<<1],Next[N<<1],head[N],E=0;
inline void add(int u,int v){
    ver[++E]=v,Next[E]=head[u],head[u]=E;
}
int ver1[N],Next1[N],head1[N],edge1[N],E1=0;
inline void add1(int u,int v,int num){
    ver1[++E1]=v,Next1[E1]=head1[u],head1[u]=E1,edge1[E1]=num;
}
int c[N<<1];
inline void addd(int x,int y){
    for(;x<=tot;x+=x&-x) c[x]+=y;
}
inline int sum(int x){
    int res=0;
    for(;x;x-=x&-x) res+=c[x];
    return res;
}
int dfn[N],sz[N],cnt=0,ans[N];
bool book[N];
void dfsfail(int u){
    dfn[u]=++cnt,sz[u]=1,book[u]=true;
    for(int i=head[u];i;i=Next[i]){
        int v=ver[i];
        if(!book[v]) dfsfail(v),sz[u]+=sz[v];
    }
}
void dfstrie(int u){
    addd(dfn[u],1);
    if(End[u]){
        for(int i=head1[End[u]];i;i=Next1[i]){
            int v=ver1[i],x=dis[v];
//            printf("%d %d %d\n",v,x,edge1[i]);
            ans[edge1[i]]=sum(dfn[x]+sz[x]-1)-sum(dfn[x]-1);
        }
    }
    for(int i=0;i<26;++i)
    if(ch[u][i]) dfstrie(ch[u][i]);
    addd(dfn[u],-1);
}
char s[N];int len,m,st;
int main(){
//    freopen("testdata.in","r",stdin);
    scanf("%s",s+1);
    len=strlen(s+1);
    for(st=1;st<=len;++st)
    if(s[st]!='B'&&s[st]!='P') break;
    int p=insert(1,s[st]-'a');
    for(int i=st+1;i<=len;++i){
        if(s[i]=='B') p=back(p);
        else if(s[i]=='P') ed(p);
        else p=insert(p,s[i]-'a');
    }
    scanf("%d",&m);
    for(int i=1,u,v;i<=m;++i){
        scanf("%d%d",&u,&v);
        add1(v,u,i);
    }
    build();
    for(int i=2;i<=tot;++i)
    add(fail[i],i),add(i,fail[i]);
    dfsfail(1),dfstrie(1);
    for(int i=1;i<=m;++i) printf("%d\n",ans[i]);
    return 0;
}