Bzoj2434 [Noi2011]阿狸的打字机

Time Limit: 10 Sec Memory Limit: 256 MB
Submit: 2536 Solved: 1415

Description

阿狸喜欢收藏各种稀奇古怪的东西，最近他淘到一台老式的打字机。打字机上只有28个按键，分别印有26个小写英文字母和'B'、'P'两个字母。

经阿狸研究发现，这个打字机是这样工作的：

l 输入小写字母，打字机的一个凹槽中会加入这个字母(这个字母加在凹槽的最后)。

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

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

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

我们把纸上打印出来的字符串从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

1 2

1 3

2 3

Sample Output

HINT

1<=N<=10^5

1<=M<=10^5

输入总长<=10^5

Source

Trie

先把字符串建成AC自动机，然后对fail指针建立DFS序，利用DFS序来差分答案。

默默打开了hzw学长的博客，默默开抄。

 #include<iostream>

 #include<cstdio>

 #include<algorithm>

 #include<cstring>

 #include<queue>

 #include<vector>

 #define LL long long

 using namespace std;

 const int mxn=;

 int read(){

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

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

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

     return x*f;

 }

 struct edge{int v,nxt;}e[mxn<<];

 int hd[mxn],mct=;

 void add_edge(int u,int v){

     e[++mct].v=v;e[mct].nxt=hd[u];hd[u]=mct;return;

 }

 vector<int>ask[mxn];//询问

 vector<int>id[mxn];

 int ans[mxn];

 //

 int n,m;

 char s[mxn];

 int ind[mxn],outd[mxn],sz=;

 int tr[mxn<<];

 void add(int x,int v){while(x<=sz){tr[x]+=v;x+=x&-x;}}

 int smm(int x){

     int res=;

     while(x){res+=tr[x];x-=x&-x;}

     return res;

 }

 int pos[mxn],sct=;

 struct Aca{

     int t[mxn][],fail[mxn];

     int fa[mxn];

     int rt,cnt;

     void clear(){cnt=;rt=;for(int i=;i<;i++)t[][i]=;}

     void insert(char s[]){

         int now=rt;

         int len=strlen(s);

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

             if(s[i]=='P'){pos[++sct]=now;continue;}

             else if(s[i]=='B'){now=fa[now];continue;}

             if(!t[now][s[i]-'a']){

                 t[now][s[i]-'a']=++cnt;

                 fa[t[now][s[i]-'a']]=now;

             }

             now=t[now][s[i]-'a'];

         }

     }

     void Build(){

         queue<int>q;

         q.push(rt);

         fail[]=;

         while(!q.empty()){

             int now=q.front();q.pop();

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

                 if(t[now][i]){

                     int k=fail[now];

                     while(!t[k][i])k=fail[k];

                     fail[t[now][i]]=t[k][i];

                     q.push(t[now][i]);

                 }

             }

         }

     }

 }aca;

 void DFS(int u){

     ind[u]=++sz;

     for(int i=hd[u];i;i=e[i].nxt){

         DFS(e[i].v);

     }

     outd[u]=++sz;

     return;

 }

 int main()

 {

     int i,j,u,v;

     scanf("%s",s);

     n=strlen(s);

     aca.clear();

     aca.insert(s);

     aca.Build();

     for(int i=aca.rt;i<=aca.cnt;i++){

         add_edge(aca.fail[i],i);

     }

     DFS();

     m=read();

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

         u=read();v=read();

         ask[v].push_back(u);

         id[v].push_back(i);

     }

     int now=aca.rt;int t=;

     //

     add(ind[],);//add(outd[1],-1);

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

 //      printf("now: %c \n",s[i]);

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

             t++;

             u=t;

             for(j=;j<ask[u].size();j++){

                 v=pos[ask[u][j]];

 //              printf("%d to %d\n",ind[v],outd[v]);

 //              printf("%d %d\n",smm(outd[v]),smm(ind[v]-1));

                 ans[id[u][j]]=smm(outd[v])-smm(ind[v]-);

             }

             continue;

         }

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

             add(ind[now],-);//add(outd[now],1);

             now=aca.fa[now];continue;

         }

         now=aca.t[now][s[i]-'a'];

         add(ind[now],);//add(outd[now],-1);

     }

     //

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

     return ;

 }