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
阿狸喜欢收藏各种稀奇古怪的东西,最近他淘到一台老式的打字机。打字机上只有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个询问的答案。
思路:
好题!
注意到询问一个串x在y中的出现次数,相当于在AC自动机的fail树上查询x的代表节点的子树中,代表y的某一个前缀的节点在其中的出现次数.
然后我们需要用到本题的一些特殊性质:每一个串都是由之前的串修改而来的,而修改不会太多.那我们可以先建出AC自动机并构造好fail树,记录DFS序,并将询问按y存进vector.
最后重新考虑开始的操作,对于新加的一个字符,走到对应的儿子节点,将其DFS序的编号加入树状数组;删除字符则走到当前节点在Trie上的父亲,将其DFS序编号从树状数组中删除;遇到'P'则考虑这个串的vector中存的询问,只需在树状数组中查询其子树DFS序的区间的和即可.
代码:
#include <cstdio>
#include <iostream>
#include <algorithm>
#include <cstring>
#include <cmath>
#include <queue>
#include <vector>
#define For(i, j, k) for(int i = j; i <= (int)k; i++)
#define Forr(i, j, k) for(int i = j; i >= (int)k; i--)
#define pb push_back
const int N = 100010;
using namespace std;
int e;
struct BIT{
int c[N];
inline int lowbit(int x){
return x & (-x);
}
void add(int x, int v){
while(x <= e + 1){
c[x] += v;
x += lowbit(x);
}
}
int sum(int x){
int ret = 0;
while(x){
ret += c[x];
x -= lowbit(x);
}
return ret;
}
}T;
int Ans[N];
vector<int> Q[N], id[N], G[N];
struct AC_Automaton{
int ch[N][26], fa[N], f[N];
int node[N], tot;
void insert(char *S){
int n = strlen(S) - 1, o = 0;
e = tot = dfn = 0;
For(i, 0, n){
if(S[i] == 'P') node[++tot] = o;
else if(S[i] == 'B') o = fa[o];
else{
int &c = ch[o][S[i] - 'a'];
if(!c) c = ++e;
fa[c] = o, o = c;
}
}
}
void getfail(){
queue<int> q;
For(i, 0, 25){
int c = ch[0][i];
if(c) f[c] = 0, q.push(c), G[0].pb(c);
}
while(!q.empty()){
int s = q.front(); q.pop();
For(i, 0, 25){
int &c = ch[s][i], o = ch[f[s]][i];
if(c){
q.push(c);
f[c] = o;
G[o].pb(c);
}
else c = o;
}
}
}
int in[N], out[N], dfn;
void DFS(int o){
in[o] = ++dfn;
For(i, 0, G[o].size() - 1) DFS(G[o][i]);
out[o] = dfn;
}
void solve(char* S){
int n = strlen(S) - 1, o = 0;
tot = 0;
For(i, 0, n){
if(S[i] == 'P'){
++tot;
For(j, 0, Q[tot].size() - 1){
int x = node[Q[tot][j]];
Ans[id[tot][j]] = T.sum(out[x]) - T.sum(in[x] - 1);
}
}
else if(S[i] == 'B') T.add(in[o], -1), o = fa[o];
else o = ch[o][S[i] - 'a'], T.add(in[o], 1);
}
}
}S;
char P[N];
int m;
int main(){
scanf("%s", P);
S.insert(P);
S.getfail();
S.DFS(0);
scanf("%d", &m);
For(i, 1, m){
int x, y;
scanf("%d%d", &x, &y);
Q[y].pb(x), id[y].pb(i);
}
S.solve(P);
For(i, 1, m) printf("%d\n", Ans[i]);
return 0;
}