bzoj 2434: [Noi2011]阿狸的打字机

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

HINT

 

 1<=N<=10^5


1<=M<=10^5

输入总长<=10^5

 

Source

Trie

 

这题乱搞的暴力分有很多。。。

首先建出AC自动机,最暴力的做法就是把y串在trie上的每个点往上跳fail,如果能跳到x的结尾则说明出现了依次;

这等价于建出fail树后,y串在trie上的点有多少个在x的子树中;

直接想法就是在fail树上链剖,把y串到根节点的路径全部加1,然后查询x的值,但这样比较慢;

标算是把路径修改单点求值 改为 单点修改子树查询,这样只要再在AC自动机上跑一边,然后把询问相同的用vector挂个链,碰到单词结尾查询一下子树和即可,可以用树状数组实现;

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
#include<vector>
using namespace std;
const int N=200050;
int trie[N][26],fa[N],fail[N],pos[N],len,tt,cnt;
int head[N*26],nxt[N*26],to[N*26],tot,flag,dfn[N],ed[N],T,c[N],ans[N];
char a[N];
void lnk(int x,int y){
	to[++tot]=y,nxt[tot]=head[x],head[x]=tot;
}
void insert(){
	int x=0;
	for(int i=1;i<=len;i++){
		if(a[i]=='P') pos[++tt]=x;
		else if(a[i]=='B') x=fa[x];
		else{
			if(!trie[x][a[i]-'a']) trie[x][a[i]-'a']=++cnt;
			fa[cnt]=x,x=trie[x][a[i]-'a'];
		}
	}
}
void build(){
	queue<int> q; fail[0]=0;
	for(int i=0;i<26;i++){
		if(trie[0][i]) q.push(trie[0][i]);
	}
	while(!q.empty()){
		int u=q.front();q.pop();
		for(int i=0;i<26;i++){
			int x=trie[u][i],j=fail[u];
			if(!x) continue;
			q.push(x);
			while(j&&!trie[j][i]) j=fail[j];
			if(trie[j][i]) fail[x]=trie[j][i];
		}
	}
	for(int i=1;i<=cnt;i++) lnk(fail[i],i);
}
void dfs(int x){
	dfn[x]=++T;
	for(int i=head[x];i;i=nxt[i]) dfs(to[i]);
	ed[x]=T;
}
void update(int x,int val){
  for(int i=x;i<=T;i+=i&-i) c[i]+=val;
}
int query(int x){
  int sum=0;
  for(int i=x;i;i-=i&-i) sum+=c[i];
  return sum;
}
struct data{
	int x,id;
};
vector<data> p[N];
void solve(){
	int m;cin>>m;
	for(int i=1;i<=m;i++){
		int x,y;scanf("%d%d",&x,&y);
		p[pos[y]].push_back((data){x,i});
	}
	int x=0;
	for(int i=1;i<=len;i++){
		if(a[i]=='P'){
			for(int j=0;j<p[x].size();j++){
				data y=p[x][j];
				ans[y.id]=query(ed[pos[y.x]])-query(dfn[pos[y.x]]-1);
			}
		}
		else if(a[i]=='B') update(dfn[x],-1),x=fa[x];
		else x=trie[x][a[i]-'a'],update(dfn[x],1);
	}
	for(int i=1;i<=m;i++) printf("%d\n",ans[i]);
}
int main(){
	scanf("%s",a+1);len=strlen(a+1);
	insert();build();dfs(0);solve();
}

  

posted @ 2017-07-13 22:00  qt666  阅读(115)  评论(0编辑  收藏  举报