Treap总结与模板

Treap

支持插入,删除,区间第k大,一个数的前驱,后继...

核心的思想:

每个节点有一个key表示该节点的值

和一个priority 表示当前节点的优先值

我们的树既满足二叉查找树的左小右大

右满足堆的上小下大

这样一来,均摊复杂度可以达到logn

在插入时,只要不满足堆的性质就旋转

在删除时,我们找到要删除的点,并将它旋转到叶子节点

在旋转时也要注意优先值

 

---

核心操作

rotate 旋转

type为0是右旋,1是左旋
void rotate(int &o,int type){
	int x=t[o].ch[type];
	t[o].ch[type]=t[x].ch[type^1];
	t[x].ch[type^1]=o; 
	t[x].size=t[o].size;
	update_size(o);
	o=x;
}

insert 插入

void insert(int &o,int val){
	if(o==0){
		o=++tot;
		t[o].size=t[o].num=1,t[o].key=val,t[o].p=randon();
		return;
	}
	t[o].size++;
	if(t[o].key==val){
		t[o].num++;return;
	}
	if(val<t[o].key){
		insert(t[o].ch[0],val);
		if(t[t[o].ch[0]].p<t[o].p)
			rotate(o,0);//右旋 
	}
	else{
		insert(t[o].ch[1],val);
		if(t[t[o].ch[1]].p<t[o].p)
			rotate(o,1);
	}
}

erase 删除

void erase(int &o,int val){
	if(o==0) return;
	if(t[o].key==val){//转到叶子节点 
		if(t[o].num>1){
			t[o].num--;
			t[o].size--;
			return;
		}
		if(t[o].ch[0]==0) o=t[o].ch[1];
		else if(t[o].ch[1]==0) o=t[o].ch[0];
		else{
			if(t[t[o].ch[0]].p<t[t[o].ch[1]].p){
				rotate(o,0),erase(o,val);
			}
			else rotate(o,1),erase(o,val);
		}
	}
	else{
		if(val<t[o].key) t[o].size--,erase(t[o].ch[0],val); 
		else t[o].size--,erase(t[o].ch[1],val);
	}
}

查询排名

int rank_x(int o,int x){//x的排名 
	if(x==t[o].key) return t[t[o].ch[0]].size+1;
	else if(x<t[o].key){
		return rank_x(t[o].ch[0],x);
	}
	else return rank_x(t[o].ch[1],x)+t[t[o].ch[0]].size+t[o].num;
}
int rank_k(int o,int k){//查排名为k的 
	if(t[t[o].ch[0]].size>=k) return rank_k(t[o].ch[0],k);
	else 
	if(t[t[o].ch[0]].size+t[o].num<k)
		return rank_k(t[o].ch[1],k-t[t[o].ch[0]].size-t[o].num);
	else return t[o].key;
}

查询前驱后继

void pre(int o,int x){
	if(o==0) return;
	if(x>t[o].key){
		ans=o,pre(t[o].ch[1],x);
	}
	else pre(t[o].ch[0],x);
}
void suf(int o,int x){
	if(o==0) return;
	if(x<t[o].key){
		ans=o,suf(t[o].ch[0],x);
	}
	else suf(t[o].ch[1],x);
}答案为t[ans].key

---

模板

#include<bits/stdc++.h>
#define N 100005
using namespace std;
struct Node{
	int ch[2],size,p,key,num;//priority
}t[N];
int n,root,ans,tot;
int read(){
    int cnt=0,f=1;char ch=0;
    while(!isdigit(ch)){ch=getchar();if(ch=='-')f=-1;}
    while(isdigit(ch))cnt=cnt*10+(ch-'0'),ch=getchar();
    return cnt*f;
}
inline int randon(){
    static int seed=233;
    return seed=int(seed*47821LL%2147483647);
}
void update_size(int o){
    t[o].size=t[o].num+t[t[o].ch[0]].size+t[t[o].ch[1]].size;
}
void rotate(int &o,int type){
	int x=t[o].ch[type];
	t[o].ch[type]=t[x].ch[type^1];
	t[x].ch[type^1]=o; 
	t[x].size=t[o].size;
	update_size(o);
	o=x;
}
void insert(int &o,int val){
	if(o==0){
		o=++tot;
		t[o].size=t[o].num=1,t[o].key=val,t[o].p=randon();
		return;
	}
	t[o].size++;
	if(t[o].key==val){
		t[o].num++;return;
	}
	if(val<t[o].key){
		insert(t[o].ch[0],val);
		if(t[t[o].ch[0]].p<t[o].p)
			rotate(o,0);//右旋 
	}
	else{
		insert(t[o].ch[1],val);
		if(t[t[o].ch[1]].p<t[o].p)
			rotate(o,1);
	}
}
void erase(int &o,int val){
	if(o==0) return;
	if(t[o].key==val){//转到叶子节点 
		if(t[o].num>1){
			t[o].num--;
			t[o].size--;
			return;
		}
		if(t[o].ch[0]==0) o=t[o].ch[1];
		else if(t[o].ch[1]==0) o=t[o].ch[0];
		else{
			if(t[t[o].ch[0]].p<t[t[o].ch[1]].p){
				rotate(o,0),erase(o,val);
			}
			else rotate(o,1),erase(o,val);
		}
	}
	else{
		if(val<t[o].key) t[o].size--,erase(t[o].ch[0],val); 
		else t[o].size--,erase(t[o].ch[1],val);
	}
}
int rank_x(int o,int x){//x的排名 
	if(x==t[o].key) return t[t[o].ch[0]].size+1;
	else if(x<t[o].key){
		return rank_x(t[o].ch[0],x);
	}
	else return rank_x(t[o].ch[1],x)+t[t[o].ch[0]].size+t[o].num;
}
int rank_k(int o,int k){//查排名为k的 
	if(t[t[o].ch[0]].size>=k) return rank_k(t[o].ch[0],k);
	else 
	if(t[t[o].ch[0]].size+t[o].num<k)
		return rank_k(t[o].ch[1],k-t[t[o].ch[0]].size-t[o].num);
	else return t[o].key;
}
void pre(int o,int x){
	if(o==0) return;
	if(x>t[o].key){
		ans=o,pre(t[o].ch[1],x);
	}
	else pre(t[o].ch[0],x);
}
void suf(int o,int x){
	if(o==0) return;
	if(x<t[o].key){
		ans=o,suf(t[o].ch[0],x);
	}
	else suf(t[o].ch[1],x);
}
int main(){
	n=read();
	for(int i=1;i<=n;i++){
		int op=read(),x=read();
		if(op==1) insert(root,x);//插入x
		if(op==2) erase(root,x);
		if(op==3) printf("%d\n",rank_x(root,x));
		if(op==4) printf("%d\n",rank_k(root,x));
		if(op==5) pre(root,x),printf("%d\n",t[ans].key);
		if(op==6) suf(root,x),printf("%d\n",t[ans].key);
	}
	return 0;
}