【NOI2011】阿狸的打字机

Description

阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有 26个小写英文字母和’B’、’P’两个字母。 经阿狸研究发现，这个打字机是这样工作的：

输入小写字母，打字机的一个凹槽中会加入这个字母(按 P 前凹槽中至少有一个字母)。 

按一下印有'B'的按键，打字机凹槽中最后一个字母会消失。 

按一下印有'P'的按键，打字机会在纸上打印出凹槽中现有的所有字母并换行，但凹槽中的字母不会消失（保证凹槽中至少有一个字母） 。 

例如，阿狸输入 aPaPBbP，纸上被打印的字符如下：

a

aa

ab

我们把纸上打印出来的字符串从 1开始顺序编号，一直到 n。打字机有一个非常有趣的功能，在打字机中暗藏一个带数字的小键盘，在小键盘上输入两个数(x,y)（其中1≤x,y≤n），打字机会显示第x个打印的字符串在第y个打印的字符串中出现了多少次。 阿狸发现了这个功能以后很兴奋，他想写个程序完成同样的功能，你能帮助他么？

1≤n≤ 10^5
1≤m≤ 10^5
输入文件第一行的字符数≤10^5

Analysis

ganxi大佬说要补专题，那就写一道AC自动机练手吧

首先，trie肯定要建出来
我们要求点x代表的串（用x简写）在y中出现了多少次
一个暴力的想法是，从root到y每个点遍历，顺着fail链往上跳，如果跳到x就累计一次
因为，fail[x]一定是x的后缀
但是时间不允许
尝试把fail树建出来
fail树中，x在所有x的子树里的点出现了一次
那么为了统计答案，只需求出某点子树内累计的点数
弄出dfs序用树状数组来搞即可

Code

#include <queue>
#include <cstdio>
#include <cstring>
#include <algorithm>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,b,a) for(int i=b;i>=a;i--)
#define efo(i,v) for(int i=last[v];i;i=next[i])
using namespace std;
typedef long long ll;
const int N=1e5+5;
int n,m,tot,now,pre[N],to[N*2],next[N*2],last[N];
int a[N],le[N],ri[N],c[N*2],ans[N];
char s[N];
queue<int> q;
struct node
{
    int x,y,id;
}b[N];
struct trie
{
    int fail,son[30];
}tr[N];
void read(int &t)
{
    t=0;char ch;
    for(ch=getchar();ch<'0' || ch>'9';ch=getchar());
    for(;'0'<=ch && ch<='9';ch=getchar()) t=t*10+ch-'0';
}
bool cmp(node a,node b){return a.y<b.y;}
void link(int u,int v)
{
    to[++tot]=v,next[tot]=last[u],last[u]=tot;
}
void getfail()
{
    q.push(1);
    while(!q.empty())
    {
        int u=q.front();q.pop();
        fo(i,0,25)
        {
            int v=tr[u].son[i];
            if(!v) continue;
            int x=tr[u].fail;
            for(;x && !tr[x].son[i];x=tr[x].fail);
            if(tr[x].son[i]) tr[v].fail=tr[x].son[i];
            else tr[v].fail=1;
            q.push(v);
        }
    }
}
void dfs(int v)
{
    le[v]=++now;
    efo(i,v) dfs(to[i]);
    ri[v]=now;
}
int lowbit(int x){return x&-x;}
void add(int x,int y)
{
    for(int i=x;i<=now;i+=lowbit(i)) c[i]+=y;
}
int query(int x)
{
    int t=0;
    for(int i=x;i;i-=lowbit(i)) t=t+c[i];
    return t;
}
int main()
{
    scanf("%s\n",s+1);
    n=strlen(s+1);
    int Q;
    read(Q);
    fo(i,1,Q) b[i].id=i,read(b[i].x),read(b[i].y);
    sort(b+1,b+Q+1,cmp);
    m=1;
    int v=1,t=0;
    fo(i,1,n)
        if(s[i]=='P') a[++t]=v;
        else
        if(s[i]=='B') v=pre[v];
        else tr[v].son[s[i]-'a']=++m,pre[m]=v,v=m;
    getfail();
    fo(i,2,m) link(tr[i].fail,i);
    dfs(1);
    int j=1;t=0;v=1;
    fo(i,1,n)
        if(s[i]=='P')
        {
            t++;
            for(;b[j].y==t;j++)
            {
                int x=a[b[j].x];
                ans[b[j].id]=query(ri[x])-query(le[x]-1);
            }
        }
        else
        if(s[i]=='B')
        {
            add(le[v],-1);
            v=pre[v];
        }
        else
        if('a'<=s[i] && s[i]<='z')
        {
            v=tr[v].son[s[i]-'a'];
            add(le[v],1);
        }
    fo(i,1,Q) printf("%d\n",ans[i]);
    return 0;
}