我的第二棵平衡树: 普通Treap

这又是洛谷平衡树的模板题...

[SBT的写法]

Treap的代码很短,只是在普通二茬查找树的插入操作和删除操作中加入维护堆性质的rotate,

形式也和堆很像,插入时将新节点不断向上抬,删除时将节点向下降直至弹出.

随机种子的选取可以用一些玄学的方法(来自网上某位大佬)

inline int random() {
	static int seed=48271;
	return seed=seed*48271LL%2147483647;
}

而rotate操作可以通过将孩子数组设为ch[2]简化成一个

void rotate(int &t,int d) {
	int x=ch[t][d^1]; ch[t][d^1]=ch[x][d]; ch[x][d]=t;
	w[x]=w[t]; w[t]=w[ch[t][0]]+w[ch[t][1]]+1;
	t=x;
}

于是代码就特别短

#include<stdio.h>
#define N 100005
int root,val[N],ch[N][2],rnd[N],w[N],size,cnt;
inline int random() {
	static int seed=48271;
	return seed=seed*48271LL%2147483647;
}
void rotate(int &t,int d) {
	int x=ch[t][d^1]; ch[t][d^1]=ch[x][d]; ch[x][d]=t;
	w[x]=w[t]; w[t]=w[ch[t][0]]+w[ch[t][1]]+1;
	t=x;
}
void insert(int &t,int v) {
	if(!t) {
		size++; t=++cnt; val[t]=v;
		rnd[t]=random(); w[t]=1; return;
	}
	w[t]++;	int d=v>val[t];
	insert(ch[t][d],v);
	if(rnd[ch[t][d]]<rnd[t]) rotate(t,d^1);
}
void erase(int &t,int v) {
	if(!t) return;
	if(v==val[t]) {
		if(!ch[t][0]) t=ch[t][1], size--;
		else if(!ch[t][1]) t=ch[t][0], size--;
		else {
			int d=rnd[ch[t][0]]<rnd[ch[t][1]];
			rotate(t,d);
			erase(ch[t][d],v);
		}
	}
	else erase(ch[t][v>val[t]],v);
	if(t) w[t]=w[ch[t][0]]+w[ch[t][1]]+1;
}
int select(int k) {
	int t=root;
	while(k!=w[ch[t][0]]+1) {
		if(k<=w[ch[t][0]]) t=ch[t][0];
		else k-=w[ch[t][0]]+1, t=ch[t][1];
	}
	return val[t];
}
int rank(int v) {
	int t=root, ret=0;
	while(t) {
		if(v<=val[t]) t=ch[t][0];
		else ret+=w[ch[t][0]]+1, t=ch[t][1];
	}
	return ret+1;
}
int pre(int v) {
	int t=root,ret=0;
	while(t) {
		if(v<=val[t]) t=ch[t][0];
		else ret=t, t=ch[t][1];
	}
	return val[ret];
}
int suc(int v) {
	int t=root,ret=0;
	while(t) {
		if(v>=val[t]) t=ch[t][1];
		else ret=t, t=ch[t][0];
	}
	return val[ret];
}
int main() {
    int n,x,y;
    scanf("%d",&n);
    while(n--) {
        scanf("%d%d",&x,&y);
        switch(x) {
            case 1: insert(root,y); break;
            case 2: erase(root,y); break;
            case 3: printf("%d\n",rank(y)); break;
            case 4: printf("%d\n",select(y)); break;
            case 5: printf("%d\n",pre(y)); break;
            case 6: printf("%d\n",suc(y)); break;
        }
    }
}