普通平衡树

您需要写一种数据结构(可参考题目标题),来维护一些数,其中需要提供以下操作:

1.插入数值 x。
2.删除数值 x(若有多个相同的数,应只删除一个)。
3.查询数值 x 的排名(若有多个相同的数,应输出最小的排名)。
4.查询排名为 x 的数值。
5.求数值 x 的前驱(前驱定义为小于 x 的最大的数)。
6.求数值 x 的后继(后继定义为大于 x 的最小的数)。
注意: 数据保证查询的结果一定存在。

输入格式
第一行为 n,表示操作的个数。

接下来 n 行每行有两个数 opt 和 x,opt 表示操作的序号(1≤opt≤6)。

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

数据范围
1≤n≤100000,所有数均在 −107 到 107 内。

输入样例:
8
1 10
1 20
1 30
3 20
4 2
2 10
5 25
6 -1
输出样例:
2
20
20
20
思路:平衡树模板
代码:

#include<stdio.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int N=1e6+10,INF=1e8+10;
int id,root;
struct l
{
	int key,val,cnt,size,l,r;
}tr[N];
void push_up(int p)
{
	tr[p].size=tr[tr[p].r].size+tr[tr[p].l].size+tr[p].cnt;
}
int get(int key)
{
	tr[++id].key=key;
	tr[id].val=rand();
	tr[id].size=tr[id].cnt=1;
	return id;
}
void build()
{
	get(-INF),get(INF);
	root=1,tr[1].r=2;
	push_up(root);
}
void right(int &p)
{
	int q=tr[p].l;
	tr[p].l=tr[q].r;
	tr[q].r=p;
	p=q;
	push_up(tr[p].r);
	push_up(p);
}
void left(int &p)
{
	int q=tr[p].r;
	tr[p].r=tr[q].l;
	tr[q].l=p;
	p=q;
	push_up(tr[p].l);
	push_up(p);
}
void insert(int &p,int key)
{
	if(!p) p=get(key);
	else if(tr[p].key==key) tr[p].cnt++;
	else if(tr[p].key>key)
	{
		insert(tr[p].l,key);
		if(tr[tr[p].l].val>tr[p].val) right(p);
	}
	else if(tr[p].key<key)
	{
		insert(tr[p].r,key);
		if(tr[tr[p].r].val>tr[p].val) left(p);
	 } 
	 push_up(p);
}
void remove(int &p,int key)
{
	if(!p) return;
	else if(tr[p].key==key)
	{
		if(tr[p].cnt>1) tr[p].cnt--;
		else if(tr[p].r||tr[p].l)
		{
			if(!tr[p].r||tr[tr[p].l].val>tr[tr[p].r].val)
			{
				right(p);
				remove(tr[p].r,key);
			}
			else
			{
				left(p);
				remove(tr[p].l,key);
			}
		}
		else p=0;
	}
	else if(tr[p].key>key) remove(tr[p].l,key);
	else if(tr[p].key<key) remove(tr[p].r,key);
	push_up(p);
}
int get_rank_by_key(int p,int key)
{
	if(!p) return 0;
	else if(tr[p].key==key) return tr[tr[p].l].size+1;
	else if(tr[p].key>key) return get_rank_by_key(tr[p].l,key);
	else return tr[tr[p].l].size+tr[p].cnt+get_rank_by_key(tr[p].r,key);
}
int get_key_by_rank(int p,int rank)
{
	if(!p) return INF;
	else if(tr[tr[p].l].size>=rank) return get_key_by_rank(tr[p].l,rank);
	else if(tr[tr[p].l].size+tr[p].cnt>=rank) return tr[p].key;
	else return get_key_by_rank(tr[p].r,rank-tr[tr[p].l].size-tr[p].cnt);
}
int get_prex(int p,int key)
{
	if(!p) return -INF;
	else if(tr[p].key>=key) return get_prex(tr[p].l,key);
	return max(tr[p].key,get_prex(tr[p].r,key));
}
int get_next(int p,int key)
{
	if(!p) return INF;
	else if(tr[p].key<=key) return get_next(tr[p].r,key);
	return min(tr[p].key,get_next(tr[p].l,key));
}
int main()
 {
 	int n,m,i,j;
 	scanf("%d",&n);
 	ll ans=0;
 	build();
 	for(i=0;i<n;i++)
 	{
 		int a,x;
 		scanf("%d",&a); 
 		if(a==1)
 		{
 			scanf("%d",&x);
 			insert(root,x);
		 }
		 if(a==2)
		 {
		 	scanf("%d",&x);
		 	remove(root,x);
		 }
		 if(a==3)
		 {
		 	scanf("%d",&x);
		 	printf("%d\n",get_rank_by_key(root,x)-1);
		 }
		 if(a==4)
		 {
		 	scanf("%d",&x);
		 	printf("%d\n",get_key_by_rank(root,x+1));
		 }
		 if(a==5)
		 {
		 	scanf("%d",&x);
		 	printf("%d\n",get_prex(root,x));
		 }
		 if(a==6)
		 {
		 	scanf("%d",&x);
		 	printf("%d\n",get_next(root,x));
		 }
	 }
	 //printf("%lld\n",ans);
 }
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值