bzoj 2434 [Noi2011]阿狸的打字机（AC自动机+fail树+dfs序+树状数组）

2434: [Noi2011]阿狸的打字机

Time Limit: 10 Sec   Memory Limit: 256 MB
Submit: 3521   Solved: 1913
[ Submit][ Status][ Discuss]

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个询问的答案。

Sample Input

aPaPBbP

3

1 2

1 3

2 3

Sample Output

2

1

0

fail树：http://blog.csdn.net/jaihk662/article/details/78155715

DFS序：http://blog.csdn.net/jaihk662/article/details/52232360

第一行输入的字符串其实就是构造Trie树的过程：

如果是小写字母，新建节点；

如果是B，回到Trie树上的父亲节点；

如果是P，说明当前节点是第y个的单词结尾，标记

要知道，字符串x在字符串y中出现多少次就是看y的所有前缀中，有多少前缀的后缀是x

而fail树中每个节点的父亲都是这个节点的最长后缀，并且每个字符串的前缀一定在fail树上有一个节点

所以可以先求出fail树，对于查询第x个单词在第y个单词中出现多少次，只要将fail树中所有y前缀节点权值设为1（其它节点权值都为0）然后查询以单词x为根的子树权值和就是答案

可是每次暴力修改节点查询子树是不可能的，那就需要离线将询问排序，然后用树状数组维护dfs序

#include<stdio.h>
#include<vector>
#include<queue>
#include<algorithm>
using namespace std;
queue<int> q;
vector<int> G[100005];
int cnt, n, tre[100005][26], fa[100005], fail[100005], word[100005];
int dcnt, dfn[100005], out[100005], bit[100005], ans[100005];
char str[100005];
typedef struct Que
{
	int id;
	int x, y;
	bool operator < (const Que &b) const
	{
		if(y<b.y)
			return 1;
		return 0;
	}
}Que;
Que s[100005];
void Update(int k, int val)
{
	while(k<=dcnt)
	{
		bit[k] += val;
		k += k&-k;
	}
}
int Query(int k)
{
	int sum = 0;
	while(k)
	{
		sum += bit[k];
		k -= k&-k;
	}
	return sum;
}
void Insert()
{
	int i, p = 1;
	for(i=1;str[i]!='\0';i++)
	{
		if(str[i]=='P')
			word[++n] = p;
		else if(str[i]=='B')
			p = fa[p];
		else
		{
			if(tre[p][str[i]-'a'+1]==0)
				tre[p][str[i]-'a'+1] = ++cnt;
			fa[tre[p][str[i]-'a'+1]] = p;
			p = tre[p][str[i]-'a'+1];
		}
	}
}
void ACmach()
{
	int p, now, i;
	fail[1] = 0;
	q.push(1);
	while(q.empty()==0)
	{
		now = q.front();
		q.pop();
		for(i=1;i<=26;i++)
		{
			if(tre[now][i]==0)
				continue;
			p = fail[now];
			while(tre[p][i]==0)
				p = fail[p];
			fail[tre[now][i]] = tre[p][i];
			q.push(tre[now][i]);
		}
	}
	for(i=2;i<=cnt;i++)
		G[fail[i]].push_back(i);
}
void Sech(int u)
{
	int i, v;
	dfn[u] = ++dcnt;
	for(i=0;i<G[u].size();i++)
	{
		v = G[u][i];
		Sech(v);
	}
	out[u] = dcnt;
}
int main(void)
{
	int i, m, now, p, id;
	cnt = 1, n = 0;
	for(i=1;i<=26;i++)
		tre[0][i] = 1;
	scanf("%s", str+1);
	Insert();
	ACmach();
	Sech(1);
	scanf("%d", &m);
	for(i=1;i<=m;i++)
	{
		scanf("%d%d", &s[i].x, &s[i].y);
		s[i].id = i;
	}
	sort(s+1, s+m+1);
	id = p = 1, now = 0;
	for(i=1;str[i]!='\0';i++)
	{
		if(str[i]=='B')
		{
			Update(dfn[p], -1);
			p = fa[p];
		}
		else if(str[i]=='P')
		{
			now++;
			while(s[id].y==now && id<=m)
			{
				ans[s[id].id] = Query(out[word[s[id].x]])-Query(dfn[word[s[id].x]]-1);
				id++;
			}
		}
		else
		{
			p = tre[p][str[i]-'a'+1];
			Update(dfn[p], 1);
		}
	}
	for(i=1;i<=m;i++)
		printf("%d\n", ans[i]);
	return 0;
}
/*
aaaaaBaBaPaaaaaaaaP
3
1 2
1 2
1 1
*/