fhq treap抄袭笔记

fhq treap

碎碎念

我咋感觉合并这么像左偏树呢
ps:难道你们的treap都是小头堆的吗
fhq真的是神人
现在看以前学的splay是有点恶心,尤其是压行压不过fhqtreap

点一下

fhq treap主要操作就俩
拆(merge)和合(split)
其他操作都是基于这俩操作(拆拆合合,合拆合拆,拆了又和,合了又拆)

合并操作merge

把两颗树合并成一颗
这里的两颗树x,y,满足x树小于y树
因为要保证堆的性质

	if(!x||!y) return x+y;//必有一颗为空,所以直接返回那颗不空树即可
	if(pri[x]<pri[y]) {//这里x为根,因为x树<y树,所以y一定在x的右孩子中
		ch[x][1]=merge(ch[x][1],y);
		pushup(x);//更新size
		return x;
	} else {//类似的,自己想
		ch[y][0]=merge(x,ch[y][0]);//这里的merge顺序千万不要颠倒,因为要x树<y树
		pushup(y);
		return y;
	}
}

拆分操作split

以一个基准数拆分
不太会说,不说了,自己去领悟吧
其他操作看这里

注意!!!

k_th的时候一定要记得去减去左边的size

模板

#include <iostream>
#include <ctime>
#include <cstdlib>
#include <cstdio>
#include <cstring>
#define FOR(i,a,b) for(int i=a;i<=b;++i)
using namespace std;
const int maxn=100007;
int read() {
	int x=0,f=1;char s=getchar();
	for(;s>'9'||s<'0';s=getchar()) if(s=='-') f=-1;
	for(;s>='0'&&s<='9';s=getchar()) x=x*10+s-'0';
	return x*f;
}
int ch[maxn][2],siz[maxn],val[maxn],pri[maxn],cnt;
void pushup(int x) {
	siz[x]=1+siz[ch[x][0]]+siz[ch[x][1]];
}
int make_edge(int x) {
	val[++cnt]=x,siz[cnt]=1,pri[cnt]=rand();
	return cnt;
}
int merge(int x,int y) {
	if(!x||!y) return x+y;
	if(pri[x]<pri[y]) {
		ch[x][1]=merge(ch[x][1],y);
		pushup(x);
		return x;
	} else {
		ch[y][0]=merge(x,ch[y][0]);
		pushup(y);
		return y;
	}
}
void split(int now,int k,int &x,int &y) {
	if(!now) x=y=0;
	else {
		if(val[now]<=k) 
			x=now,split(ch[now][1],k,ch[x][1],y);
		else 
			y=now,split(ch[now][0],k,x,ch[y][0]);
		pushup(now);
	}
}
int k_th(int now,int k) {
	while(233) {
		if(k==siz[ch[now][0]]+1) return now;
		if(k<=siz[ch[now][0]]) now=ch[now][0];
		else k-=siz[ch[now][0]]+1,now=ch[now][1];
	}
}
int main() {
	int n=read(),rt=0;
	FOR(i,1,n) {
		int opt=read(),a=read(),x,y,z;
		if(opt==1){
			split(rt,a,x,y);
			rt=merge(merge(x,make_edge(a)),y);
		} else if(opt==2) {
			split(rt,a,x,z);
			split(x,a-1,x,y);
			y=merge(ch[y][0],ch[y][1]);
			rt=merge(merge(x,y),z);
		} else if(opt==3) {
			split(rt,a-1,x,y);
			printf("%d
",siz[x]+1);
			rt=merge(x,y);
		} else if(opt==4) {
			printf("%d
",val[k_th(rt,a)]);
		} else if(opt==5) {
			split(rt,a-1,x,y);
			printf("%d
",val[k_th(x,siz[x])]);
			rt=merge(x,y);
		} else if(opt==6) {
			split(rt,a,x,y);
			printf("%d
",val[k_th(y,1)]);
			rt=merge(x,y);
		}
	}
	return 0;
}
原文地址:https://www.cnblogs.com/dsrdsr/p/10041027.html