BZOJ1861: [Zjoi2006]Book 书架 Splay

BZOJ1861: [Zjoi2006]Book 书架

Time Limit: 4 Sec   Memory Limit: 64 MB

Submit: 1302   Solved: 747

[Submit][Status][Discuss]

题解：Splay

对于两种询问操作不多说。

对于剩下的几种修改，本质上是一模一样的，首先要判断如果是Insert而且T==0那么就continue

然后先把x点从树中删除，具体删除方法是把区间[x,x]Splay出来然后把ls[x+1]断开

然后就是插入，把所要找的位置的区间[x,x+1]splay出来，此时ls[x+1]为empty，然后直接把节点接到x+1就可以了（其实对于Bottom和Top可以简化一些代码见）

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
int ls[N],rs[N],siz[N],fa[N],a[N],n,m,root,S,T;
char c[11];
void Pushup(int x)
{
	siz[x]=siz[ls[x]]+siz[rs[x]]+1;
}
void print()
{
	cout<<"root="<<root<<endl;
	for(int i=0;i<=n+2;i++) printf("i=%d ls=%d rs=%d fa=%d siz=%d\n",i,ls[i],rs[i],fa[i],siz[i]);
} 
void Rotate(int x,int &k)
{
    int y=fa[x],z=fa[y];
    if(k==y) k=x;
    else
    {
        if(y==ls[z]) ls[z]=x;
        else rs[z]=x;
    }
    fa[x]=z;
    fa[y]=x;
    if(ls[y]==x)
    {
        ls[y]=rs[x];
        fa[rs[x]]=y;
        rs[x]=y;
    }
    else
    {
        rs[y]=ls[x];
        fa[ls[x]]=y;
        ls[x]=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(ls[z]==y^ls[y]==x) Rotate(x,k);
			else Rotate(y,k);
		}
		Rotate(x,k);
	}
}
void Build(int l,int r,int f)
{
	if(l>r) return;
	if(l==r)
	{
		siz[a[l]]=1;
		fa[a[l]]=a[f];
		if(l<f) ls[a[f]]=a[l];
		else rs[a[f]]=a[l];
		return;
	}
	int mid=(l+r)>>1;
	Build(l,mid-1,mid);
	Build(mid+1,r,mid);
	Pushup(a[mid]);
	fa[a[mid]]=a[f];
	if(mid<f) ls[a[f]]=a[mid];
	else rs[a[f]]=a[mid];
}
int tot; 
int Getnum(int x,int rank)
{
	if(rank==siz[ls[x]]+1) return x;
	else if(rank<=siz[ls[x]]) return Getnum(ls[x],rank);
	else return Getnum(rs[x],rank-siz[ls[x]]-1);
}
int Getrank(int x)
{
	Splay(x,root);
	return siz[ls[x]]+1;
}
void Del(int x)
{
	int tmp=Getrank(x);
	int y=Getnum(root,tmp-1),z=Getnum(root,tmp+1);
	Splay(y,root);
	Splay(z,rs[y]);
	ls[z]=fa[x]=0;
	Pushup(z);
}
void Ins(int nu,int x)
{
	int u=Getnum(root,nu);
	int v=Getnum(root,nu+1);
	Splay(u,root);
	Splay(v,rs[u]);
	ls[v]=x;
	fa[x]=v;
	Pushup(v);
}
void Top(int x)
{
	int u=Getnum(root,2);
	Splay(u,root);
	fa[S]=x,ls[x]=S;
	fa[x]=u,ls[u]=x;
	Pushup(S),Pushup(x),Pushup(u);
}
void Bot(int x)
{
	int u=Getnum(root,n);
	Splay(u,root);
	fa[T]=x,rs[x]=T;
	fa[x]=u,rs[u]=x;
	Pushup(T),Pushup(x),Pushup(u);
}
int main()
{
	scanf("%d%d",&n,&m);
	for(int i=2;i<=n+1;i++) scanf("%d",&a[i]);
	S=n+1,T=n+2;
	a[1]=S,a[n+2]=T;
	root=a[(n+3)>>1];
	Build(1,n+2,0);
	int x,y,u,v;
	while(m--)
	{
		scanf("%s%d",c,&x);
		if(c[0]=='Q') printf("%d\n",Getnum(root,x+1));
		else if(c[0]=='A') printf("%d\n",Getrank(x)-2);
		else if(c[0]=='I')
		{
			scanf("%d",&y);
			if(y==0) continue;
			int tmp=Getrank(x);
			Del(x);
			if(y==-1) Ins(tmp-2,x);
			if(y==1) Ins(tmp,x);
		}
		else
		{
			Del(x);
			if(c[0]=='T') Top(x);
			if(c[0]=='B') Bot(x);
		}
	}
}

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

[Submit][Status][Discuss]