P2596 [ZJOI2006]书架（Splay）

题目描述

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

输入格式

第一行有两个整数，分别表示书的个数 n 以及命令条数 m。
第二行有 n 个整数，第 i 个整数表示初始时从上向下书第 i 本书的编号 pi 。
接下来 m 行，每行表示一个操作。每行初始时有一个字符串 op。
若 op 为 Top，则后有一个整数 s，表示把编号为 s 的书放在最上面。
若 op 为 Bottom，则后有一个整数 s，表示把编号为 s 的书放在最下面。
若 op 为 Insert，则后有两个整数 s, t，表示若编号为 s 的书上面有 x 本书，则放回这本书时他的上面有 x + t 本书。
若 op 为 Ask，则后面有一个整数 s，表示询问编号为 s 的书上面有几本书。
若 op 为 Query，则后面有一个整数 s，询问从上面起第 s 本书的编号。
输出格式
对于每次查询，输出一行一个整数表示答案。

输入输出样例

输入
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
输出
2
9
9
7
5
3

说明/提示

数据规模与约定
对于 100% 的数据，保证：3≤n,m≤8×10⁴。
pi是一个 1∼n 的排列。
1≤s≤n，−1≤t≤1，op 只可能是输入的五种字符串之一。
当编号为 s 的书上面没有书的时候，不会对它进行 Insert s -1 操作。
当编号为 s 的书下面没有书的时候，不会对它进行 Insert s 1 操作。

题目分析

一道Splay的模板题，题目中的五个操作分别对应：

Top：首先将s节点旋转到根，然后将s节点的左子树插到s后继节点的左子树上。

Bottom：首先将s节点旋转到根，然后将s节点的右子树插到s前驱节点的右子树上。

Insert：如果t=1，则交换s与其后继节点即可；t=-1则交换s与其前驱节点。

Ask：首先将s节点旋转到根，输出其左子树的大小即可。

Query：直接查询第s名的节点即可（模板）。

代码如下

#include <iostream>
#include <cmath>
#include <cstdio>
#include <set>
#include <unordered_set>
#include <string>
#include <cstring>
#include <map>
#include <unordered_map>
#include <algorithm>
#include <stack>
#include <queue>
#include <bitset>
#define LL long long
#define PII pair<int,int>
#define PDD pair<double,double>
#define x first
#define y second
using namespace std;
const int N=1e5+5,INF=0x3f3f3f3f;
struct Node{
	int s[2],v,p;
	int size;
}tr[N];
int pos[N];				//pos[]记录该权值对应的节点编号
int root,idx;
void pushup(int u)
{
	tr[u].size=tr[tr[u].s[0]].size+tr[tr[u].s[1]].size+1;
	pos[tr[tr[u].s[0]].v]=tr[u].s[0];
	pos[tr[tr[u].s[1]].v]=tr[u].s[1];
}
void rotate(int x)
{
	int y=tr[x].p,z=tr[y].p;
	int k=tr[y].s[1]==x;
	tr[z].s[tr[z].s[1]==y]=x,tr[x].p=z;
	tr[y].s[k]=tr[x].s[k^1],tr[tr[x].s[k^1]].p=y;
	tr[x].s[k^1]=y,tr[y].p=x;
	pushup(y),pushup(x);
}
void splay(int x,int k)
{
	while(tr[x].p!=k)
	{
		int y=tr[x].p,z=tr[y].p;
		if(z!=k)
			if((tr[z].s[1]==y)^(tr[y].s[1]==x)) rotate(x);
			else rotate(y);
		rotate(x);
	}
	if(!k) root=x;
}
void insert(int v)			//插入一个节点
{
	pos[v]=++idx;
	tr[idx].size=1;
	tr[idx].s[0]=tr[idx].s[1]=0;
	tr[idx].v=v;
	if(idx>1)
	{
		tr[idx-1].s[1]=idx;
		tr[idx].p=idx-1;
		pushup(idx-1);
		splay(idx,0);
	}
}
int query(int u,int k)		//查询第k小的数
{
	if(tr[tr[u].s[0]].size>=k) return query(tr[u].s[0],k);
	else if(tr[tr[u].s[0]].size+1==k) return u;
	else return query(tr[u].s[1],k-tr[tr[u].s[0]].size-1);
}
void top(int s)			//首先将s节点旋转到根，然后将s节点的左子树插到s后继节点的左子树上。
{
	s=pos[s];
	splay(s,0);
	if(!tr[s].s[0]) return;
	if(!tr[s].s[1]) tr[s].s[1]=tr[s].s[0],tr[s].s[0]=0;
	else 
	{
		int u=query(root,tr[tr[s].s[0]].size+2);
		tr[u].s[0]=tr[s].s[0];
		tr[tr[s].s[0]].p=u;
		tr[s].s[0]=0;
		splay(u,0);
	}
}
void bottom(int s)		//首先将s节点旋转到根，然后将s节点的右子树插到s前驱节点的右子树上。
{
	s=pos[s];
	splay(s,0);
	if(!tr[s].s[1]) return;
	if(!tr[s].s[0]) tr[s].s[0]=tr[s].s[1],tr[s].s[1]=0;
	else 
	{
		int u=query(root,tr[tr[s].s[0]].size);
		tr[u].s[1]=tr[s].s[1];
		tr[tr[s].s[1]].p=u;
		tr[s].s[1]=0;
		splay(u,0);
	}
}
void update(int s,int t)	//如果t=1，则交换s与其后继节点即可；t=-1则交换s与其前驱节点。
{
	if(!t) return;			//t==0则不需要进行操作
	s=pos[s];
	splay(s,0);
	int x;
	if(t==1) x=query(root,tr[tr[s].s[0]].size+2);
	else x=query(root,tr[tr[s].s[0]].size);
	swap(pos[tr[s].v],pos[tr[x].v]);
	swap(tr[s].v,tr[x].v);
}
int ask(int s)		//首先将s节点旋转到根，输出其左子树的大小即可。
{
	s=pos[s];
	splay(s,0);
	return tr[tr[s].s[0]].size;
}
int main()
{
	int n,m;
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
	{
		int x;
		scanf("%d",&x);
		insert(x);
	}
	while(m--)
	{
		char op[10];
		int s,t;
		scanf("%s%d",op,&s);
		switch(op[0])
		{
			case 'T': top(s); break;
			case 'B': bottom(s); break;
			case 'I':
				scanf("%d",&t); 
				update(
				s,t); break;
			case 'A': printf("%d\n",ask(s)); break;
			case 'Q': printf("%d\n",tr[query(root,s)].v); break;
		}
	}
	return 0;
}