BZOJ 2434: [Noi2011]阿狸的打字机 [AC自动机 Fail树树状数组 DFS序]

2434: [Noi2011]阿狸的打字机

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2545 Solved: 1419
[Submit][Status][Discuss]

Description

阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有26个小写英文字母和'B'、'P'两个字母。
经阿狸研究发现，这个打字机是这样工作的：
l 输入小写字母，打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。
l 按一下印有'B'的按键，打字机凹槽中最后一个字母会消失。
l 按一下印有'P'的按键，打字机会在纸上打印出凹槽中现有的所有字母并换行，但凹槽中的字母不会消失。
例如，阿狸输入aPaPBbP，纸上被打印的字符如下：
a
aa
ab
我们把纸上打印出来的字符串从1开始顺序编号，一直到n。打字机有一个非常有趣的功能，在打字机中暗藏一个带数字的小键盘，在小键盘上输入两个数(x,y)（其中1≤x,y≤n），打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。
阿狸发现了这个功能以后很兴奋，他想写个程序完成同样的功能，你能帮助他么？

Input

输入的第一行包含一个字符串，按阿狸的输入顺序给出所有阿狸输入的字符。

第二行包含一个整数m，表示询问个数。

接下来m行描述所有由小键盘输入的询问。其中第i行包含两个整数x, y，表示第i个询问为(x, y)。

Output

输出m行，其中第i行包含一个整数，表示第i个询问的答案。

Sample Input

aPaPBbP
3
1 2
1 3
2 3

Sample Output

2
1
0

HINT

1<=N<=10^5

1<=M<=10^5

输入总长<=10^5

煞笔错误毁我青春

先想了想一个字符串建一个AC自动机，不行啊会爆的，因为B只能删除一个

诶，这个打字机好有意思，只能删一个的话顺着模拟下来就可以了把Trie树建出来啊，建一个AC自动机就好了【需要维护fa，别忘写了】

然后....这也是模板就是文本，和BZOJ3127挺像，用类似的想法，统计Fail树中x的子树里有多少个y的节点，就是x在y中的出现次数了

但是，怎么统计y的出现次数？想到这就不会了，看题解，好神啊

求Fail树的dfs序(显式建图)，然后din[x]和dout[x]之间的序列就是x的子树，转换成序列问题

统计序列一段区间y用到的节点出现次数

神奇的打字机性质，发现每个y也都是模拟打字过程中的某个时间的结果，考虑离线，按y的小到大排序，依次模拟，用树状数组维护当前模拟到的字符的出现次数：

遇到新字符 add(新字符的dfs序，1)遇到B add(当前字符的dfs序，-1)

遇到P 看看是不是到了询问(注意多个询问y相同的情况)，到了的话查询pos[x]的进出dfs序之间的和，就是y的出现次数了

pos[x]是第x个模板的Trie节点

查询复杂度O(nlogn)

注意：

1.你加的双向边，所以dfs判!=fa

2.中途改数组名改齐全了，别有的地方没改结果找半天

3.树状数组维护的是dfs序，大小为dfc，不是n，也不是sz(sz是从0(root=0)开始的，应该比dfc少1)

#include <iostream>

#include <cstdio>

#include <cstring>

#include <algorithm>

using namespace std;

const int N=1e5+;

inline int read(){

    char c=getchar();int x=,f=;

    while(c<''||c>''){if(c=='-')f=-; c=getchar();}

    while(c>=''&&c<=''){x=x*+c-''; c=getchar();}

    return x*f;

}

int n,m;

char s[N];

struct ques{

    int x,y,i;

    bool operator <(const ques &r)const {return y<r.y;}

}a[N];

struct edge{

    int v,ne;

}e[N<<];

int cnt=,h[N];

inline void ins(int u,int v){//printf("ins %d %d\n",u,v);

    cnt++;

    e[cnt].v=v;e[cnt].ne=h[u];h[u]=cnt;

    cnt++;

    e[cnt].v=u;e[cnt].ne=h[v];h[v]=cnt;

}

int din[N],dout[N],dfc;

void dfs(int u,int fa){

    din[u]=++dfc;

    for(int i=h[u];i;i=e[i].ne) if(e[i].v!=fa) dfs(e[i].v,u);

    dout[u]=dfc;//printf("dfs %d %d %d\n",u,din[u],dout[u]);

}

struct node{

    int ch[],fa,fail;

}t[N];

int sz,pos[N],tot;

void build(char s[]){

    int u=;

    for(int i=;i<=n;i++){

        if(s[i]=='B') u=t[u].fa;

        else if(s[i]=='P') pos[++tot]=u;

        else{

            int c=s[i]-'a';

            if(!t[u].ch[c]) t[u].ch[c]=++sz;

            t[t[u].ch[c]].fa=u;

            u=t[u].ch[c];

        }

    }

    //for(int i=0;i<=sz;i++) printf("build %d %d %d\n",i,t[i].fa,t[i].ch[0]);

}

int q[N],head,tail;

void getFail(){

    head=tail=;

    for(int i=;i<;i++) if(t[].ch[i])

        q[tail++]=t[].ch[i],ins(,t[].ch[i]);//!!!ins

    while(head!=tail){

        int u=q[head++];

        for(int i=;i<;i++){

            int &v=t[u].ch[i];

            if(!v) {v=t[t[u].fail].ch[i];continue;}

            t[v].fail=t[t[u].fail].ch[i];

            q[tail++]=v;

            ins(t[v].fail,v);

        }

    }

}

int c[N];

inline int lowbit(int x){return x&-x;}

inline void add(int p,int v){for(int i=p;i<=dfc;i+=lowbit(i))c[i]+=v;}//,printf("[add %d\n",i);}

inline int sum(int p){

    int re=;

    for(int i=p;i;i-=lowbit(i)) re+=c[i];//,printf("sum %d %d\n",i,c[i]);

    return re;

}

int ans[N];

void solve(){

    sort(a+,a++m);

    build(s);

    getFail();

    dfs(,-);

    int u=,p=,num=;

    for(int i=;i<=n;i++){//printf("hi %d %c\n",i,s[i]);

        if(s[i]=='B') add(din[u],-),u=t[u].fa;

        else if(s[i]=='P'){

            num++;

            while(num==a[p].y){

                int l=din[pos[a[p].x]],r=dout[pos[a[p].x]];

                int t1=sum(l-),t2=sum(r);//printf("que %d  %d %d  %d %d\n",a[p].i,l,r,t1,t2);

                ans[a[p].i]=t2-t1;

                p++;

            }

        }else{

            int c=s[i]-'a';

            u=t[u].ch[c]; //printf("hehe %d %d\n",u,din[u]);

            add(din[u],);

        }

        //printf("u %d\n",u);

    }

    for(int i=;i<=m;i++) printf("%d\n",ans[i]);

}

int main(){

    //freopen("in.txt","r",stdin);

    scanf("%s",s+);m=read();

    n=strlen(s+);

    for(int i=;i<=m;i++) a[i].x=read(),a[i].y=read(),a[i].i=i;

    solve();

}