普通平衡树【treap模板】

题目描述

链接

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

• 插入 $x$ 数
• 删除 $x$ 数 ( 若有多个相同的数，因只删除一个 )
• 查询 $x$ 数的排名 ( 排名定义为比当前数小的数的个数 $+1$ )
• 查询排名为 $x$ 的数
• 求 $x$ 的前驱 ( 前驱定义为小于 $x$ ，且最大的数 )
• 求 $x$ 的后继 ( 后继定义为大于 $x$ ，且最小的数 )

输入格式

第一行为 $n$ ，表示操作的个数,下面 $n$ 行每行有两个数 $\text{opt}$ 和 $x$，$\text{opt}$ 表示操作的序号 ( $1\le \text{opt}\le 6$ )

输出格式

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

输入输出样例

输入

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

输出

106465
84185
492737

数据范围

对于 $100\%$ 的数据，$1\le n\le 10^5$ ，$|x|\le 10^7$.

思路

可以用 treap 来实现上述功能 。

同学的这篇博客总结的很好，可以参考参考 ( treap 笔记 )。其实是我不想总结

treap 不严格要求平衡，由于随机数的影响，树的最大深度难以预测，所以需要动态分配结点。貌似大家都懂

#include<bits/stdc++.h>
using namespace std;
const int inf=0x7fffffff,N=1e5+10;
struct Treap{ int l,r,val,cnt,size; unsigned dat; } a[N];
int tot,root,op,val,m;

signed qrand(){
    static signed seed=703;
    return seed=seed*48271;
}

int New(int val){
	a[++tot].val=val;
	a[tot].cnt=a[tot].size=1;
	a[tot].dat=qrand();
	return tot;
}

void update(int p){
	a[p].size=a[a[p].l].size+a[a[p].r].size+a[p].cnt;
}

void build(){
	New(-inf),New(inf);
	root=1,a[1].r=2;
	update(root);
}

void zip(int &p){
	int q=a[p].l;
	a[p].l=a[q].r,a[q].r=p,p=q;
	update(a[p].r),update(p);
}

void zap(int &p){
	int q=a[p].r;
	a[p].r=a[q].l,a[q].l=p,p=q;
	update(a[p].l),update(p);
}

int getRankByVal(int p,int val){
	if(p==0) return 0;
  	if(val==a[p].val) return a[a[p].l].size+1;
	if(val<a[p].val) return getRankByVal(a[p].l,val);
	return getRankByVal(a[p].r,val)+a[a[p].l].size+a[p].cnt;
}

int getValByRank(int p,int rk){
	if(p==0) return inf;
	if(a[a[p].l].size>=rk) return getValByRank(a[p].l,rk);
	if(a[a[p].l].size+a[p].cnt>=rk) return a[p].val;
	return getValByRank(a[p].r,rk-a[a[p].l].size-a[p].cnt);
}

void insert(int &p,int val){
	if(p==0) return (void)(p=New(val));
	if(val==a[p].val) return (void)(a[p].cnt++,update(p));
	if(val<a[p].val){
		insert(a[p].l,val);
		if(a[p].dat<a[a[p].l].dat) zip(p);
	}else{
		insert(a[p].r,val);
		if(a[p].dat<a[a[p].r].dat) zap(p);
	}
	update(p);
}

int getNext(int val){
	int ans=2,p=root;
	while(p){
		if(val==a[p].val){
			if(a[p].r>0){
				p=a[p].r;
				while(a[p].l>0) p=a[p].l;
				ans=p;
			}
			break;
		}
		if(a[p].val>val&&a[p].val<a[ans].val) ans=p;
		p=val<a[p].val?a[p].l:a[p].r;
	}
	return a[ans].val;
}

int getPre(int val){
	int ans=1,p=root;
	while(p){
		if(val==a[p].val){
			if(a[p].l>0){
				p=a[p].l;
				while(a[p].r>0) p=a[p].r;
				ans=p;
			}
			break;
		}
		if(a[p].val<val&&a[p].val>a[ans].val) ans=p;
		p=val<a[p].val?a[p].l:a[p].r;
	}
	return a[ans].val;
}

void remove(int &p,int val){
	if(p==0) return;
	if(val==a[p].val){
		if(a[p].cnt>1) return (void)(a[p].cnt--,update(p));
		if(a[p].l||a[p].r){
			if(a[p].r==0||a[a[p].l].dat>a[a[p].r].dat)
				zip(p),remove(a[p].r,val);
			else zap(p),remove(a[p].l,val);
			update(p);
		}else p=0;
		return;
	}
	val<a[p].val?remove(a[p].l,val):remove(a[p].r,val);
	update(p);
}

int main(){
	ios::sync_with_stdio(false);
	build();
	for(cin>>m;m;m--){
		cin>>op>>val;
		switch(op){
			case 1: insert(root,val); break;
			case 2: remove(root,val); break;
			case 3: cout<<getRankByVal(root,val)-1<<"\n"; break;
			case 4: cout<<getValByRank(root,val+1)<<"\n"; break;
			case 5: cout<<getPre(val)<<"\n"; break;
			case 6: cout<<getNext(val)<<"\n"; break;
		}
	}
}