Treap 求调
查看原帖
Treap 求调
134066
Pethly_Cat楼主2022/6/12 10:39
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+5;
int root,tot;
struct Treap{
	int l,r,val,dat,cnt,size;
}a[N];
int New(int val){
	a[++tot].val=val,a[tot].dat=rand(),a[tot].cnt=1,a[tot].size=1;
	return tot;
}
void pushup(int p){
	a[p].size=a[a[p].l].size+a[a[p].r].size+a[p].cnt;
}
void build(){
	root=New(-2e9),a[root].r=New(2e9);
	pushup(root);
}
void zig(int &p){
	int q=a[p].l;
	a[p].l=a[q].r,a[q].r=p,p=q;
	pushup(a[p].r); pushup(p);
}
void zag(int &p){
	int q=a[p].r;
	a[p].r=a[q].l,a[q].l=p,p=q;
	pushup(a[p].l); pushup(p);
}
void insert(int &p,int val){
	if(p==0){
		p=New(val);
		return;
	}
	if(val==a[p].val){
		a[p].cnt++;
		return;
	}
	if(val<a[p].val){
		insert(a[p].l,val);
		if(a[p].dat<a[a[p].l].dat) zig(p);
	}
	else{
		insert(a[p].r,val);
		if(a[p].dat<a[a[p].r].dat) zag(p);
	}
	pushup(p);
}
void remove(int &p,int val){
	if(p==0) return;
	if(val==a[p].val){
		if(a[p].cnt>1){
			a[p].cnt--; pushup(p);
			return;
		}
		if(a[p].l||a[p].r){
			if(a[p].r==0||a[a[p].l].dat>a[a[p].r].dat)
				zig(p),remove(a[p].r,val);
			else zag(p),remove(a[p].l,val);
			pushup(p);
		}
		else p=0;
		return;
	}
	val<a[p].val? remove(a[p].l,val):remove(a[p].r,val);
	pushup(p);
}
int getrank(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 getrank(a[p].l,val);
	return getrank(a[p].r,val)+a[a[p].l].size+a[p].cnt;
}
int getval(int p,int val){
	if(p==0) return 2e9;
	if(a[a[p].l].size>=val) return getval(a[p].l,val);
	if(a[a[p].l].size+a[p].cnt>=val) return a[p].val;
	return getval(a[p].r,val-a[a[p].l].size-a[p].cnt);
}
int getpre(int val){
	int p=root,ans=1;
	while(p){
		if(a[p].val<val) ans=a[p].val,p=a[p].r;
		else p=a[p].l;
	}
	return ans;
}
int getnext(int val){
	int p=root,ans=1;
	while(p){
		if(a[p].val>val) ans=a[p].val,p=a[p].l;
		else p=a[p].r;
	}
	return ans;
}
int main(){
	int T; scanf("%d",&T);
	build();
	while(T--){
		int op,x; scanf("%d%d",&op,&x);
		if(op==1) insert(root,x);
		if(op==2) remove(root,x);
		if(op==3) printf("%d\n",getrank(root,x)-1);
		if(op==4) printf("%d\n",getval(root,x+1));
		if(op==5) printf("%d\n",getpre(x));
		if(op==6) printf("%d\n",getnext(x));
	}
	return 0;
}
2022/6/12 10:39
加载中...