fhq treap抄袭笔记

时间:2023-11-24 21:34:50

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\n",siz[x]+1);
rt=merge(x,y);
} else if(opt==4) {
printf("%d\n",val[k_th(rt,a)]);
} else if(opt==5) {
split(rt,a-1,x,y);
printf("%d\n",val[k_th(x,siz[x])]);
rt=merge(x,y);
} else if(opt==6) {
split(rt,a,x,y);
printf("%d\n",val[k_th(y,1)]);
rt=merge(x,y);
}
}
return 0;
}