BZOJ 1861 [Zjoi2006] Book 书架

Description

小T有一个很大的书柜。这个书柜的构造有些独特，即书柜里的书是从上至下堆放成一列。她用1到n的正整数给每本书都编了号。 小T在看书的时候，每次取出一本书，看完后放回书柜然后再拿下一本。由于这些书太有吸引力了，所以她看完后常常会忘记原来是放在书柜的什么位置。不过小T的记忆力是非常好的，所以每次放书的时候至少能够将那本书放在拿出来时的位置附近，比如说她拿的时候这本书上面有X本书，那么放回去时这本书上面就只可能有X-1、X或X+1本书。 当然也有特殊情况，比如在看书的时候突然电话响了或者有朋友来访。这时候粗心的小T会随手把书放在书柜里所有书的最上面或者最下面，然后转身离开。 久而久之，小T的书柜里的书的顺序就会越来越乱，找到特定的编号的书就变得越来越困难。于是她想请你帮她编写一个图书管理程序，处理她看书时的一些操作，以及回答她的两个提问：(1)编号为X的书在书柜的什么位置；(2)从上到下第i本书的编号是多少。

Input

第一行有两个数n，m，分别表示书的个数以及命令的条数；第二行为n个正整数：第i个数表示初始时从上至下第i个位置放置的书的编号；第三行到m+2行，每行一条命令。命令有5种形式： 1． Top S——表示把编号为S的书房在最上面。 2． Bottom S——表示把编号为S的书房在最下面。 3． Insert S T——T∈{-1，0，1}，若编号为S的书上面有X本书，则这条命令表示把这本书放回去后它的上面有X+T本书； 4． Ask S——询问编号为S的书的上面目前有多少本书。 5． Query S——询问从上面数起的第S本书的编号。

Output

对于每一条Ask或Query语句你应该输出一行，一个数，代表询问的答案。

Sample Input

Sample Output

HINT

数据范围

Source

Day2

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

splay~

好久没有写了……错误百出……简直要吐血了……

事实证明黄学长的代码永远是最简洁的……

#include<cstdio>#include<cstring>#include<iostream>using namespace std;#define inf 999999999int n,m,pos[80005],a[80005],root,c[80005][2],v[80005],fa[80005],siz[80005];char s[10];void update(int u){	siz[u]=siz[c[u][0]]+siz[c[u][1]]+1;}void rotate(int x,int &k){	int y=fa[x],z=fa[y],l,r;	if(c[y][0]==x) l=0;	else l=1;	r=l^1;	if(y==k) k=x;	else if(c[z][0]==y) c[z][0]=x;	else c[z][1]=x;	fa[x]=z;fa[y]=x;fa[c[x][r]]=y;	c[y][l]=c[x][r];c[x][r]=y;	update(y);update(x);}void splay(int x,int &k){	while(x!=k)	{		int y=fa[x],z=fa[y];		if(y!=k)		{			if(x==c[y][0]^y==c[z][0]) rotate(x,k);			else rotate(y,k);		}		rotate(x,k);	}}void build(int l,int r,int u){	if(l>r) return;	int now=l,last=u;	if(l==r)	{		siz[now]=1;v[now]=a[now];fa[now]=last;		if(now<last) c[last][0]=now;		else c[last][1]=now;return;	}	int mid=(l+r)>>1;now=mid;	build(l,mid-1,mid);build(mid+1,r,mid);	v[now]=a[now];update(now);fa[now]=last;	if(mid<u) c[u][0]=mid;	else c[u][1]=mid;}int findd(int u,int rank){	int l=c[u][0],r=c[u][1];	if(siz[l]+1==rank) return u;	else if(siz[l]>=rank) return findd(l,rank);	else return findd(r,rank-siz[l]-1);}void del(int k){	int x=findd(root,k-1),y=findd(root,k+1);	splay(x,root);splay(y,c[x][1]);	int z=c[y][0];c[y][0]=0;fa[z]=siz[z]=0;	update(y);update(x);}void move(int k,int val){	int x,y,z=pos[k],rank;	splay(z,root);rank=siz[c[z][0]]+1;	del(rank);	if(val==inf) x=findd(root,n),y=findd(root,n+1);	else if(val==-inf) x=findd(root,1),y=findd(root,2);	else x=findd(root,rank+val-1),y=findd(root,rank+val);	splay(x,root);splay(y,c[x][1]);	siz[z]=1;fa[z]=y;c[y][0]=z;	update(y);update(x);}int main(){	scanf("%d%d",&n,&m);int x,y;	for(int i=2;i<=n+1;i++) scanf("%d",&a[i]),pos[a[i]]=i;	build(1,n+2,0);root=(n+3)>>1;	while(m--)	{		scanf("%s%d",s,&x);		switch(s[0])		{			case'T':move(x,-inf);break;			case'B':move(x,inf);break;			case'I':scanf("%d",&y);move(x,y);break;			case'A':splay(pos[x],root);PRintf("%d/n",siz[c[pos[x]][0]]-1);break;			case'Q':printf("%d/n",v[findd(root,x+1)]);break;		}	}	return 0;}