BZOJ_2434_[Noi2011]阿狸的打字机_AC自动机+出栈入栈序+树状数组

BZOJ_2434_[Noi2011]阿狸的打字机_AC自动机+出栈入栈序+树状数组

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

---

可以发现那个字符串相当于建立了一棵Trie树。

x在y中出现的次数等价于有多少个y串上的结点能通过fail指针走若干次能够到达x的终止节点。

也即x的终止节点在fail树上的子树中有多少个在y串上的结点。

于是我们需要处理“到子树路径加”“子树求和”。

先把fail树建出出栈入栈序，然后查询挂链，把x挂到y上，用给出的字符串遍历这颗Trie树。

向儿子走时入栈位置+1，向父亲走时出栈位置-1，打印的时候处理询问，对应子树区间为入栈到出栈这一段区间。

 

代码：

#include <cstdio>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 100050
struct Query {
    int head[N],to[N],nxt[N],cnt,val[N];
    inline void add(int u,int v,int w) {
        to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt; val[cnt]=w;
    }
}A;
struct Fail_Tree {
    int head[N],to[N],nxt[N],cnt;
    inline void add(int u,int v) {
        to[++cnt]=v; nxt[cnt]=head[u]; head[u]=cnt;
    }
}B;
int ch[N][26],fail[N],Q[N],l,r,m,cnt=1,siz[N],fa[N],dfn[N],son[N],c[N<<1],ans[N],flg[N];
char w[N];
void build() {
    int i,p;
    for(i=0;i<26;i++) ch[0][i]=1;
    Q[r++]=1;
    while(l<r) {
        p=Q[l++];
        for(i=0;i<26;i++) {
            if(ch[p][i]) fail[ch[p][i]]=ch[fail[p]][i],Q[r++]=ch[p][i];
            else ch[p][i]=ch[fail[p]][i];
        }
    }   
}
void dfs(int x) {
    int i;dfn[x]=++dfn[0];
    for(i=B.head[x];i;i=B.nxt[i]) {
        dfs(B.to[i]);
    }
    son[x]=++dfn[0];
}
void fix(int x,int v) {for(;x<=dfn[0];x+=x&(-x)) c[x]+=v;}
int inq(int x) {int re=0;for(;x;x-=x&(-x)) re+=c[x]; return re;}
void solve() {
    int i,p=1,j;
    for(i=1;w[i];i++) {
        if(w[i]=='B') {
            fix(son[p],-1);
            p=fa[p];
        }else if(w[i]=='P') {
            for(j=A.head[p];j;j=A.nxt[j]) {
                // printf("%d\n",A.val[j]);
                ans[A.val[j]]=inq(son[A.to[j]])-inq(dfn[A.to[j]]-1);
            }
        }else {
            p=ch[p][w[i]-'a'];
            fix(dfn[p],1);
        }
    }
}
int main() {
    scanf("%s%d",w+1,&m);
    int i,p=1,x,y;
    int bajisbdbdbioabsd=0;
    for(i=1;w[i];i++) {
        if(w[i]=='B') {
            p=fa[p];
        }else if(w[i]=='P') {
            siz[p]++;
            flg[++bajisbdbdbioabsd]=p;
        }else {
            int &k=ch[p][w[i]-'a'];
            if(!k) k=++cnt; fa[k]=p; p=k;
        }
    }
    build();
    for(i=1;i<=cnt;i++) B.add(fail[i],i);
    dfs(1);
    // printf("%d\n",B.cnt);puts("FUCK");
    // for(i=1;i<=cnt;i++) printf("%d %d\n",dfn[i],son[i]);
    for(i=1;i<=m;i++) scanf("%d%d",&x,&y),A.add(flg[y],flg[x],i);
    solve();
    for(i=1;i<=m;i++) printf("%d\n",ans[i]);
}