NOI 2011 阿狸的打字机 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个询问的答案。

思路：

好题！

注意到询问一个串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;
}