【bzoj2434】 Noi2011—阿狸的打字机

http://www.lydsy.com/JudgeOnline/problem.php?id=2434 (题目链接)

题意

　　给出一个字符串，$P$表示输出，$B$表示退格。$m$组询问$(x,y)$，问第$x$个串在第$y$个串中出现了多少次。

Solution

　　构出fail树，搞出dfs序，那么显然问题就转化为了自动机上匹配y时经过的节点有多少个在x的子树内，我们将询问排好序做过去树状数组单点修改维护子树和就可以了。

细节

　　mdzz建ACM和查询的时候没注意，直接重新开始TLE飞了。

代码

// bzoj2434
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
#include<ctime>
#define LL long long
#define inf (1ll<<30)
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout)
using namespace std;

const int maxn=100010;
int n,m,sz=1,cnt,dfn,pos[maxn],head[maxn],c[maxn],in[maxn],out[maxn],ans[maxn];
char ch[maxn],tmp[maxn];
struct data {int x,y,id;}q[maxn];
struct edge {int to,next;}e[maxn<<1];
struct node {
	int son[26],next,fa;
	int& operator [] (int x) {return son[x];}
}tr[maxn];

int lowbit(int x) {
	return x&-x;
}
void add(int x,int val) {
	for (int i=x;i<=sz;i+=lowbit(i)) c[i]+=val;
}
int query(int x) {
	int res=0;
	for (int i=x;i;i-=lowbit(i)) res+=c[i];
	return res;
}
void link(int u,int v) {
	e[++cnt]=(edge){v,head[u]};head[u]=cnt;
	e[++cnt]=(edge){u,head[v]};head[v]=cnt;
}
void dfs(int x,int fa) {
	in[x]=++dfn;
	for (int i=head[x];i;i=e[i].next)
		if (e[i].to!=fa) dfs(e[i].to,x);
	out[x]=dfn;
}
bool cmp(data a,data b) {
	return a.y<b.y;
}
void buildfail() {
	queue<int> q;q.push(1);
	tr[1].next=0;
	while (!q.empty()) {
		int x=q.front();q.pop();
		for (int i=0;i<26;i++) if (tr[x][i]) {
				int p=tr[x].next;
				while (!tr[p][i]) p=tr[p].next;
				link(tr[x][i],tr[p][i]);
				tr[tr[x][i]].next=tr[p][i];
				q.push(tr[x][i]);
			}
	}
}
int main() {
	scanf("%s",ch+1);
	for (int i=0;i<26;i++) tr[0][i]=1;
	n=strlen(ch+1);int p=1,id=0;
	for (int i=1;i<=n;i++) {
		if (ch[i]=='P') pos[++id]=p;
		else if (ch[i]=='B') p=tr[p].fa;
		else {
			if (!tr[p][ch[i]-'a']) tr[p][ch[i]-'a']=++sz;
			tr[tr[p][ch[i]-'a']].fa=p;p=tr[p][ch[i]-'a'];
		}
	}
	buildfail();
	dfs(1,0);
	scanf("%d",&m);
	for (int i=1;i<=m;i++) scanf("%d%d",&q[i].x,&q[i].y),q[i].id=i;
	sort(q+1,q+1+m,cmp);
	int pp=1;id=0,p=1;
	for (int i=1;i<=n;i++) {
		if (ch[i]=='P') {
			id++;
			for (;q[pp].y==id;pp++) {
				int t=pos[q[pp].x];
				ans[q[pp].id]=query(out[t])-query(in[t]-1);
			}
		}
		else if (ch[i]=='B') add(in[p],-1),p=tr[p].fa;
		else p=tr[p][ch[i]-'a'],add(in[p],1);
	}
	for (int i=1;i<=m;i++) printf("%d\n",ans[i]);
	return 0;
}