bzoj1861: [Zjoi2006]Book 书架

1861: [Zjoi2006]Book 书架

Time Limit: 4 Sec   Memory Limit: 64 MB
Submit: 1306   Solved: 751
[ Submit][ Status][ Discuss]

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

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

Sample Output

2
9
9
7
5
3

HINT

数据范围

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

 

Source

Day2

插入，删除，查询操作，裸的平衡树，我写的splay

随便搞可过

代码：

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstdlib>
#include<cmath>
#include<cstring>
#include<string>
#include<climits>
#include<queue>
#include<stack>
#include<map>
#include<set>
#define N 200010
using namespace std;
int read()
{
    int x=0,f=1;char ch;
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m,a[N],v[N],id[N*10];
int tr[N][2],fa[N],size[N];
int rt;
void pushup(int x)
{
    int l=tr[x][0],r=tr[x][1];
    size[x]=size[l]+size[r]+1;
}
void rotate(int x,int &k)
{
    int y=fa[x],z=fa[y],l,r;
    if(tr[y][0]==x)l=0;else l=1;r=l^1;
    if(y==k)k=x;
    else tr[z][tr[z][1]==y]=x;
    fa[x]=z,fa[y]=x,fa[tr[x][r]]=y;
    tr[y][l]=tr[x][r],tr[x][r]=y;
    pushup(y);pushup(x);
}
void splay(int x,int &k)
{
    while(x!=k)
    {
		int y=fa[x],z=fa[y];
		if(y!=k)
		{
		    if(tr[y][0]==x^tr[z][0]==y)
			rotate(x,k);
		    else rotate(y,k);
		}rotate(x,k);
    }
}
int find(int k,int rk)
{
    int l=tr[k][0],r=tr[k][1];
    if(rk<=size[l])return find(l,rk);
    else if(rk==size[l]+1)return k;
    else return find(r,rk-size[l]-1);
}
void new_node(int k)
{
    v[k]=a[k-1];
    size[k]=1;
}
int build(int l,int r)
{
    if(l>r)return 0;int mid=(l+r)>>1;new_node(mid);
    tr[mid][0]=build(l,mid-1);
	fa[tr[mid][0]]=mid;
    tr[mid][1]=build(mid+1,r);
	fa[tr[mid][1]]=mid;
    pushup(mid);return mid;
}
void insert(int x,int rk)
{
    int k=id[x];
    splay(k,rt);int rank=size[tr[rt][0]];
    int t1=find(rt,rank);
    int t2=find(rt,rank+2);
    splay(t1,rt);splay(t2,tr[t1][1]);
    tr[t2][0]=0;splay(t2,rt);
    t1=find(rt,rk);t2=find(rt,rk+1);
    splay(t1,rt);splay(t2,tr[rt][1]);
    tr[t2][0]=k;size[k]=1,v[k]=x;fa[k]=t2;
    splay(k,rt);
}
int main()
{
    n=read(),m=read();
    for(int i=1;i<=n;i++)
    {
		a[i]=read();
		id[a[i]]=i+1;
    }
    build(1,n+2);rt=(n+3)>>1;
    char ch[10];
    for(int i=1;i<=m;i++)
    {
		scanf("%s",ch);
		int x=read(),y;
		if(ch[0]=='T')insert(x,1);
		else if(ch[0]=='B')insert(x,n);
		else if(ch[0]=='I')
		{
			splay(id[x],rt);
			int rank=size[tr[rt][0]];
		    y=read(),insert(x,rank+y);
		}
		else if(ch[0]=='A')
		{
			splay(id[x],rt);
			printf("%d\n",size[tr[rt][0]]-1);
		}
		else printf("%d\n",v[find(rt,x+1)]);
    }
}