Treap[BZOJ3224]普通平衡树

最新推荐文章于 2019-09-09 22:57:45 发布

原创最新推荐文章于 2019-09-09 22:57:45 发布 · 215 阅读

0 ·

CC 4.0 BY-SA版权

模板同时被 2 个专栏收录

6 篇文章

订阅专栏

平衡树

1 篇文章

订阅专栏

本文对比了Treap与Splay两种数据结构，并分享了一种简洁高效的Treap实现方式。通过79行代码解决了实际问题，相比Splay的复杂度有所降低。文章详细介绍了Treap的插入、删除、查找等操作，并提供了完整的C++代码示例。

写了发treap，比splay好写多了2333
有个dalao的博客。。。日常安利,代码写的很得数组党的心。。。
不算丧病的压行79行结束战斗，比splay动不动上百行友好多了。。。
我的代码和dalao的有一点不一样就是rotate的时候我的是把左旋和右旋写到一起，因为左右旋其实是对称的，写到一起就行了，就写一个数组 $t.ch[2]$ 表示左右儿子。
还有，因为嫌弃rand()常数大，所以我打了一个随机数的表，或许能港记。

/**************************************************************
    Problem: 3224
    User: 568877201
    Language: C++
    Result: Accepted
    Time:456 ms
    Memory:38796 kb
****************************************************************/

#include<iostream>
#include<cstdio>
/*Treap*/
int n,opt,p,ans,rt;
const int N=50005;
struct Treap {
    int ch[2],val,siz,cnt,prio;//ls ch[0],rs ch[1]
} t[N<<5];
int tot,sz,rd[30]= {18333,29820,10757,21866,11757,6209,12761,32459,8312,24791,12980,15005,18195,7950,8507,13765,15014,27005,21944,21302,20191,3945,17258,22014,9256,13032,14880,27782,5591};
void pushup(int x) {t[x].siz=t[t[x].ch[0]].siz+t[t[x].ch[1]].siz+t[x].cnt;}
void rotate(int &x,int d) {
    int y=t[x].ch[d^1];
    t[x].ch[d^1]=t[y].ch[d];t[y].ch[d]=x;t[y].siz=t[x].siz;
    pushup(x);x=y;
}
void insert(int &x,int k) {
    if(x==0) {
        sz++;x=sz;t[x].val=k;t[x].prio=rd[tot++];
        if(tot==20)tot=0;t[x].siz=t[x].cnt=1;
        return;
    }
    ++t[x].siz;
    if(t[x].val==k) t[x].cnt++;
    else if(k>t[x].val) {
        insert(t[x].ch[1],k);
        if(t[t[x].ch[1]].prio<t[x].prio) rotate(x,0);
    } else {
        insert(t[x].ch[0],k);
        if(t[t[x].ch[0]].prio<t[x].prio) rotate(x,1);
    }
}
void del(int &x,int k) {
    if(x==0) return;
    if(t[x].val==k) {
        if(t[x].cnt>1) {t[x].cnt--,t[x].siz--;return;}
        if(t[x].ch[0]*t[x].ch[1]==0) x=t[x].ch[0]+t[x].ch[1];
        else if(t[t[x].ch[0]].prio<t[t[x].ch[1]].prio) rotate(x,1),del(x,k);
        else rotate(x,0),del(x,k);
    }
    else if(k>t[x].val) --t[x].siz,del(t[x].ch[1],k);
    else --t[x].siz,del(t[x].ch[0],k);
}
int GetRank(int x,int k){
    if(x==0)return 0;
    if(t[x].val==k) return t[t[x].ch[0]].siz+1;
    else if(k>t[x].val) return t[t[x].ch[0]].siz+t[x].cnt+GetRank(t[x].ch[1],k);
    else return GetRank(t[x].ch[0],k);
}
int GetVal(int x,int k){
    if(x==0) return 0;
    if(k<=t[t[x].ch[0]].siz) return GetVal(t[x].ch[0],k);
    else if(k>t[t[x].ch[0]].siz+t[x].cnt) return GetVal(t[x].ch[1],k-t[x].cnt-t[t[x].ch[0]].siz);
    else return t[x].val;   
}
void pre(int x,int k){
    if(x==0)return ;
    if(t[x].val<k){ans=x;pre(t[x].ch[1],k);}
    else pre(t[x].ch[0],k);
}
void nxt(int x,int k){
    if(!x)return;
    if(t[x].val>k){ans=x;nxt(t[x].ch[0],k);}
    else nxt(t[x].ch[1],k);
}
int main() {
    std::cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%d%d",&opt,&p);
        switch (opt){
            case 1:insert(rt,p);break;
            case 2: del(rt,p);break;
            case 3:printf("%d\n",GetRank(rt,p));break;
            case 4:printf("%d\n",GetVal(rt,p));break;
            case 5:pre(rt,p);printf("%d\n",t[ans].val);ans=0;break;
            case 6:nxt(rt,p);printf("%d\n",t[ans].val);ans=0;break;
        }
    }
    return 0;
}