Splay模板(BZOJ3224)

224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 8140   Solved: 3431
[ Submit][ Status][ Discuss]

Description

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

Input

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)

Output

对于操作3,4,5,6每行输出一个数，表示对应答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

HINT

1.n的数据范围：n<=100000

2.每个数的数据范围：[-1e7,1e7]

数据如下http://pan.baidu.com/s/1jHMJwO2

Source

平衡树

[ Submit][ Status][ Discuss]
这算是初学Splay的模板题了，代码量爆炸，争取尽快背过（理解最好）；

#include<bits/stdc++.h>
using namespace std;
const int maxn = 100010;
int fa[maxn],son[maxn][2];
int siz[maxn],val[maxn],cnt[maxn];
int n;
int rt,tot;
int next,prev;
void updata(int x)
{
	siz[x]=siz[son[x][0]]+siz[son[x][1]]+cnt[x];
}
int find(int a,int x)
{
	if(a==0) return 0;
	if(val[a]==x) return a;
	if(x>val[a]) return find(son[a][1],x);
	else return find(son[a][0],x);
	updata(a);
}
int pre(int x)
{
	int y=son[x][0];
	while(son[y][1]) y=son[y][1];
	return y;
}
void rotate(int x)
{
	int y=fa[x],z=fa[y];
	int l,r;
	if(x==son[y][0]) l=0;else l=1;r=l^1;
	if(z)
	   if(y==son[z][0]) son[z][0]=x;
	   else son[z][1]=x;
	fa[x]=z;fa[y]=x;fa[son[x][r]]=y;
	son[y][l]=son[x][r];son[x][r]=y;
	updata(y);
}
void splay(int x,int tar)
{
	int y,z;
	while(fa[x]!=tar)
	{
		y=fa[x];z=fa[y];
		if(z!=tar) if((x==son[y][0])^(y==son[z][0])) rotate(x);else rotate(y);
		rotate(x);
	}
	if(tar==0) rt=x;
	updata(x);
}
void insert(int a,int x)
{
	if(rt==0)
	{
		tot++;
		rt=tot;
		val[tot]=x;
		siz[tot]=cnt[tot]=1;
		return ;
	}
	int y;
	while(1)
	{
		if(val[a]==x) 
		{
			cnt[a]++,updata(a);break;	
		}
		int wh;
		if(x>val[a]) wh=1;else wh=0;
		y=son[a][wh];
		if(y==0)
		{
			tot++;
			cnt[tot]=siz[tot]=1;
			val[tot]=x;
			fa[tot]=a;
			son[a][wh]=tot;
			a=tot;
			break;
		}
		a=y;
	}
	splay(a,0);
}
void del(int x)
{
	int k=find(rt,x);
	if(k==0) return ;
	x=k;
	splay(x,0);
	if(cnt[x]>1){cnt[x]--,updata(x);return;} 
	if(son[x][0]==0&&son[x][1]==0){rt=0;return;}
	if(son[x][0]==0){rt=son[x][1];fa[son[x][1]]=0;return ;}
	if(son[x][1]==0){rt=son[x][0];fa[son[x][0]]=0;return ;}
	int y=pre(x);
	splay(y,x);rt=y;
	son[y][1]=son[x][1];fa[y]=0;fa[son[x][1]]=y;updata(y);
}
int rank(int x)
{
	int ans=0,now=rt;
	while(now)
	{
		if(val[now]==x) return ans+1+siz[son[now][0]];
		else if(val[now]<x) ans+=cnt[now]+siz[son[now][0]],now=son[now][1];
		else now=son[now][0];
	}
	return ans;
}
int kth(int a,int x)
{
	if(a==0) return 0;
	if(siz[son[a][0]]>=x) return kth(son[a][0],x);
	else if(siz[son[a][0]]+cnt[a]>=x) return val[a];
	else return kth(son[a][1],x-siz[son[a][0]]-cnt[a]);
}
void lb(int a,int x)
{
	if(a==0) return ;
	if(val[a]<x) prev=val[a],lb(son[a][1],x);
	if(val[a]>=x) lb(son[a][0],x);
}
void ub(int a,int x)
{
	if(a==0) return ;
	if(val[a]>x) next=val[a],ub(son[a][0],x);
	if(val[a]<=x) ub(son[a][1],x);
}
int rlr(int x)
{
	if(son[x][0]) rlr(son[x][0]);
	printf("%d ",val[x]);
	if(son[x][1]) rlr(son[x][1]);
}
int main()
{
	scanf("%d",&n);
	while(n--)
	{
		int flag,x;
		scanf("%d%d",&flag,&x);
		if(flag==1) insert(rt,x);
		if(flag==2) del(x);
		if(flag==3) printf("%d\n",rank(x));
		if(flag==4) printf("%d\n",kth(rt,x));
		if(flag==5) lb(rt,x),printf("%d\n",prev);
		if(flag==6) ub(rt,x),printf("%d\n",next);
	}
	return 0;
}

神犇们已经超我一大截了，我不求赶超，只求紧随其后。

一个小发现： 关于rank操作，网上模板大部分是上面的写法，不好记忆，需要临时推，我最近YY出了一种好写的形式：直接找到想查询的节点，Splay到根，答案就是她的左儿子siz；不知道对不对或者有没有细节问题，但是OJ评测是对的。 代码如下：

int rank(int x)
{
   int y=find(rt,x);
   splay(y,0);
   return siz[son[y][0]];
}