題目描述

阿貍喜歡收藏各種稀奇古怪的東西，最近他淘到一臺老式的打字機。打字機上只有28個按鍵，分別印有26個小寫英文字母和‘B‘、‘P‘兩個字母。

經阿貍研究發現，這個打字機是這樣工作的：

l 輸入小寫字母，打字機的一個凹槽中會加入這個字母(這個字母加在凹槽的最後)。

l 按一下印有‘B‘的按鍵，打字機凹槽中最後一個字母會消失。

l 按一下印有‘P‘的按鍵，打字機會在紙上打印出凹槽中現有的所有字母並換行，但凹槽中的字母不會消失。

例如，阿貍輸入aPaPBbP，紙上被打印的字符如下：

a
aa
ab

我們把紙上打印出來的字符串從1開始順序編號，一直到n。打字機有一個非常有趣的功能，在打字機中暗藏一個帶數字的小鍵盤，在小鍵盤上輸入兩個數(x,y)（其中1≤x,y≤n），打字機會顯示第x個打印的字符串在第y個打印的字符串中出現了多少次。

輸入

輸入的第一行包含一個字符串，按阿貍的輸入順序給出所有阿貍輸入的字符。

第二行包含一個整數m，表示詢問個數。

接下來m行描述所有由小鍵盤輸入的詢問。其中第i行包含兩個整數x, y，表示第i個詢問為(x, y)。

輸出

輸出m行，其中第i行包含一個整數，表示第i個詢問的答案。

樣例輸入

樣例輸出

提示

1<=N<=10^5

  1 #include<cmath>
  2 #include<queue>
  3 #include<vector>
  4 #include<cstdio>
  5 #include<cstring>
  6 #include<cstdlib>
  7 #include<iostream>
  8 #include<algorithm>
  9 using namespace std;
 10 queue<int>q;
 11 int m;
 12 int x,y;
 13 int cnt;
 14 int tot;
 15 int num;
 16 int len;
 17 int cont;
 18 int l[100010];//dfs序上每個數的子樹對應區間的左端點
 19 int r[100010];//dfs序上每個數的子樹對應區間的右端點
 20 int t[100010];//樹狀數組
 21 int f[100010];//每個串的終止節點
 22 char s[100010];//讀入字符串
 23 int to[100010];
 24 int fa[100010];//trie樹上每個點的父親
 25 int ans[100010];//答案
 26 int val[100010];
 27 int fail[100010];//失配指針
 28 int next[100010];
 29 int head[100010];
 30 int a[100010][26];
 31 void add(int x,int y,int v)
 32 {
 33     tot++;
 34     next[tot]=head[x];
 35     head[x]=tot;
 36     to[tot]=y;
 37     val[tot]=v;
 38 }
 39 int ask(int x)
 40 {
 41     int res=0;
 42     for(int i=x;i;i-=i&-i)
 43     {
 44         res+=t[i];
 45     }
 46     return res;
 47 }
 48 void change(int x,int v)
 49 {
 50     for(int i=x;i<=num;i+=i&-i)
 51     {
 52         t[i]+=v;
 53     }
 54 }
 55 void build(char *s)//建立trie樹
 56 {
 57     int now=0;
 58     for(int i=0;i<len;i++)
 59     {
 60         if(s[i]==‘B‘)
 61         {
 62             now=fa[now];
 63         }
 64         else if(s[i]==‘P‘)
 65         {
 66             f[++cont]=now;
 67         }
 68         else
 69         {
 70             a[now][s[i]-‘a‘]=++cnt;
 71             fa[cnt]=now;
 72             now=a[now][s[i]-‘a‘];
 73         }
 74     }
 75 }
 76 void getfail()//找每個點失配標記
 77 {
 78     q.push(0);
 79     while(!q.empty())
 80     {
 81         int now=q.front();
 82         q.pop();
 83         if(now!=0)
 84         {
 85             add(fail[now],now,0);
 86         }
 87         for(int i=0;i<26;i++)
 88         {
 89             if(a[now][i]!=0)
 90             {
 91                 q.push(a[now][i]);
 92                 if(now!=0)
 93                 {
 94                     fail[a[now][i]]=a[fail[now]][i];
 95                 }
 96                 else
 97                 {
 98                     fail[a[now][i]]=0;
 99                 }
100             }
101             else
102             {
103                 a[now][i]=a[fail[now]][i];
104             }
105         }
106     }
107 }
108 void dfs(int x)//找fail樹的dfs序
109 {
110     l[x]=++num;
111     for(int i=head[x];i;i=next[i])
112     {
113         dfs(to[i]);
114     }
115     r[x]=num;
116 }
117 int main()
118 {
119     scanf("%s",&s);
120     len=strlen(s);
121     build(s);
122     getfail();
123     dfs(0);
124     scanf("%d",&m);
125     tot=0;
126     memset(head,0,sizeof(head));
127     for(int i=1;i<=m;i++)
128     {
129         scanf("%d%d",&x,&y);
130         add(y,x,i);
131     }
132     int now=0;
133     cont=0;
134     for(int i=0;i<len;i++)
135     {
136         if(s[i]-‘a‘==-31)
137         {
138             change(l[now],-1);
139             now=fa[now];
140         }
141         else if(s[i]-‘a‘==-17)
142         {
143             for(int j=head[++cont];j;j=next[j])
144             {
145                 ans[val[j]]=ask(r[f[to[j]]])-ask(l[f[to[j]]]-1);
146             }
147         }
148         else
149         {
150             now=a[now][s[i]-‘a‘];
151             change(l[now],1);
152         }
153     }
154     for(int i=1;i<=m;i++)
155     {
156         printf("%d\n",ans[i]);
157     }
158 }

BZOJ2434[Noi2011]阿貍的打字機——AC自動機+dfs序+樹狀數組