[ZJOI2006]书架

[ZJOI2006]书架

题目描述

小T有一个很大的书柜。这个书柜的构造有些独特，即书柜里的书是从上至下堆放成一列。她用1到n的正整数给每本书都编了号。

小T在看书的时候，每次取出一本书，看完后放回书柜然后再拿下一本。由于这些书太有吸引力了，所以她看完后常常会忘记原来是放在书柜的什么位置。不过小T的记忆力是非常好的，所以每次放书的时候至少能够将那本书放在拿出来时的位置附近，比如说她拿的时候这本书上面有X本书，那么放回去时这本书上面就只可能有X-1、X或X+1本书。

当然也有特殊情况，比如在看书的时候突然电话响了或者有朋友来访。这时候粗心的小T会随手把书放在书柜里所有书的最上面或者最下面，然后转身离开。

久而久之，小T的书柜里的书的顺序就会越来越乱，找到特定的编号的书就变得越来越困难。于是她想请你帮她编写一个图书管理程序，处理她看书时的一些操作，以及回答她的两个提问：(1)编号为X的书在书柜的什么位置；(2)从上到下第i本书的编号是多少。

输入格式：

第一行有两个数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本书的编号。

输出格式：

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

输入样例#1： 

10 101 3 2 7 5 8 10 4 9 6Query 3Top 5Ask 6Bottom 3Ask 3Top 6Insert 4 -1Query 5Query 2Ask 2

输出样例#1：  

299753

说明

100%的数据，n,m <= 80000

题解:

其实就是几个splay的基本操作，不算难。

像这种码农题（其实这还算好的）思路必须先构筑清晰在写，不然调试很难受。

代码：

#include<bits/stdc++.h>
using namespace std;

const int max_n = 80010;
const int inf   = 1e9+7;

char s[15];
int n,q,x,y;

struct splay_node
{
	int size,num;
	splay_node *ch[2],*pre;
	void set_son(splay_node *son,int wh);
	void update();
	int  get_wh(){return this==pre->ch[0]? 0 : 1;}
}pool[max_n],*null,*root;

void splay_node::update()
{
	size=ch[0]->size+ch[1]->size+1;
	if(num==0) size-=1;
}

void splay_node::set_son(splay_node *son,int wh)
{
	ch[wh]=son;;
	if(son!=null) son->pre=this;
	update();
}

inline void init()
{
	null=pool+max_n-2;
	null->size=null->num=0;
	null->ch[0]=null->ch[1]=null->pre=null;
	root=null;
}

inline splay_node *newnode(int x)
{
	splay_node *newone=pool+x;
	newone->size=1;
	if(x==0) newone->size=0;
	newone->num=x;
	newone->ch[0]=newone->ch[1]=newone->pre=null;
	return newone;
}

inline void rotate(splay_node *now)
{
	splay_node *fa=now->pre,*grand=now->pre->pre;
	int wh=now->get_wh();
	fa->set_son(now->ch[wh^1],wh);
	now->set_son(fa,wh^1);
	now->pre=grand;
	if(grand!=null) grand->ch[grand->ch[0]==fa ? 0 : 1]=now;
	fa->update(); now->update();
	if(grand!=null) grand->update();
}

inline void splay(splay_node *now,splay_node *tar)
{
	for(; now->pre!=tar; rotate(now))
	  if(now->pre->pre!=tar) 
	    now->get_wh()==now->pre->get_wh() ? rotate(now->pre) : rotate(now);
	if(tar==null) root=now;
}

inline splay_node *find(int x)
{
	splay_node *now=root;
	int tot=0;
	while(now!=null)
	{
		int mor=now->size-now->ch[1]->size;
		if(tot+mor==x) return now;
		if(tot+mor>x) now=now->ch[0];
		else tot+=mor,now=now->ch[1]; 
	}
	return null;
}

inline int kth(int x)
{
	splay_node *now=pool+x;
	splay(now,null);
	return now->size-now->ch[1]->size;
}

inline void del(int x)
{
	splay_node *l=find(x-1),*r=find(x+1);
	splay(l,null); splay(r,l);
	r->set_son(null,0); l->update();
}

inline void ins(splay_node *now,int x)
{
	splay_node *l=find(x-1),*r=find(x);
	splay(l,null); splay(r,l);
	r->set_son(now,0); l->update();
}

inline void top()
{
	scanf("%d",&x);
	splay_node *now=pool+x;
	del(kth(x)); ins(now,1);
}

inline void bottom()
{
	scanf("%d",&x);
	splay_node *now=pool+x;
	del(kth(x)); ins(now,n);
}

inline void insert()
{
	scanf("%d%d",&x,&y);
	splay_node *now=pool+x;
	int k=kth(x);
	del(k); ins(now,k+y);
}

inline void ask()
{
	scanf("%d",&x);
	printf("%d\n",kth(x)-1);
}

inline void query()
{
	scanf("%d",&x);
	splay_node *now=find(x);
	printf("%d\n",now->num);
}

inline void print()
{
	for(int i=1; i<=n; ++i)
	  printf("%d ",find(i)->num);
	printf("\n");
}

int main()
{
	scanf("%d%d",&n,&q);
	init();
	splay_node *last=newnode(0);
	for(int i=1; i<=n; ++i)
	{
		scanf("%d",&x);
		splay_node *now=newnode(x);
		last->set_son(now,1);
		last=now; splay(last,null);
	}
	splay_node *now=newnode(n+1);
	last->set_son(now,1);
	last=now; splay(last,null);
	for(int i=1; i<=q; ++i)
	{
		scanf("%s",s);
		switch(s[0])
		{
			case'T': top();    break;
			case'B': bottom(); break;
			case'I': insert(); break;
			case'A': ask();    break;
			case'Q': query();  break;	
		}
//		print();
	}
	return 0;
}