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;
}