BZOJ 3224 Tyvj 1728 普通平衡树——treap

3224: Tyvj 1728 普通平衡树

Description
您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)
Input
第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)
Output
对于操作3,4,5,6每行输出一个数，表示对应答案
解题思路
裸的treap，但是本人蒟蒻刚学完treap激动地想贴一下代码（指针还不是很懂，呵呵呵）。我们都知道treap是tree（二叉排序树）+heap（二叉堆），就是满足堆性质的二叉排序树，因为二叉排序树不稳（比如退化成链），所以有人想到随机一个数，使其满足堆性质，这样只有极低的概率会被卡住（几乎不可能），期望效率（log（n））。因为其他人有专门写treap的blog写的很好（比如邻居zzk神犇），我这里并不是想写解法。对于这道题大家要注意一下有负数。

#include<cstdio>
#include<cstdlib>
using namespace std;
inline int _read(){
    int sum=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9') {if (ch=='-') f=-f;ch=getchar();}
    while (ch>='0'&&ch<='9') sum=sum*10+ch-48,ch=getchar();
    return sum*f;
}
struct jz{
    int s,x,w,l,r,ran;
}a[100005];
int n,ro,ans,tot;
void Updata(int k){a[k].s=a[a[k].l].s+a[a[k].r].s+a[k].w;}
void rturn(int &k){
    int t=a[k].l;a[k].l=a[t].r;a[t].r=k;
    a[t].s=a[k].s;Updata(k);k=t;
}
void lturn(int &k){
    int t=a[k].r;a[k].r=a[t].l;a[t].l=k;
    a[t].s=a[k].s;Updata(k);k=t;
}
void Insert(int &k,int x){
    if (k==0){k=++tot;a[k].x=x;a[k].s=a[k].w=1;a[k].ran=rand();return;}
    a[k].s++;if (a[k].x==x) a[k].w++;
    else if (x<a[k].x){
        Insert(a[k].l,x);
        if (a[a[k].l].ran<a[k].ran) rturn(k);
    }else{
        Insert(a[k].r,x);
        if (a[a[k].r].ran<a[k].ran) lturn(k);
    }
}
void del(int &k,int x){
    if (k==0) return;
    if (a[k].x==x){
        if (a[k].w>1){a[k].w--;a[k].s--;return;}else
        if (a[k].l*a[k].r==0) k=a[k].l+a[k].r;else
        if (a[a[k].l].ran<a[a[k].r].ran){rturn(k);del(k,x);}//这里原来写成了大根堆，深感抱歉，感谢CHNWJD大神的指正
        else {lturn(k);del(k,x);}
    }else if (x<a[k].x) a[k].s--,del(a[k].l,x);else a[k].s--,del(a[k].r,x);
}
int ask1(int k,int x){
    if (k==0) return 0;
    if (a[k].x==x) return a[a[k].l].s+1;
    else if (x<a[k].x) return ask1(a[k].l,x);
    else return a[a[k].l].s+a[k].w+ask1(a[k].r,x);
}
int ask2(int k,int x){
    if (k==0) return 0;
    if (x<=a[a[k].l].s) return ask2(a[k].l,x);
    else if (x>a[a[k].l].s+a[k].w) return ask2(a[k].r,x-a[a[k].l].s-a[k].w);
    else return a[k].x;
}
void pre(int k,int x){
    if (k==0) return;
    if (a[k].x<x){ans=a[k].x;pre(a[k].r,x);}else pre(a[k].l,x);
}
void sub(int k,int x){
    if (k==0) return;
    if (a[k].x>x){ans=a[k].x;sub(a[k].l,x);}else sub(a[k].r,x);
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    n=_read();
    while (n--){
        int y=_read(),x=_read();
        switch(y){
            case 1:Insert(ro,x);break;
            case 2:del(ro,x);break;
            case 3:printf("%d\n",ask1(ro,x));break;
            case 4:printf("%d\n",ask2(ro,x));break;
            case 5:ans=0,pre(ro,x),printf("%d\n",ans);break;
            case 6:ans=0,sub(ro,x),printf("%d\n",ans);break;
        }
    }
    return 0;
}

我良（zhong）心（yu）发（xue）现（hui）指针版，感谢各位dalao指正

#include<cstdio>
#include<cstdlib>
using namespace std;
const int maxn=100005;
struct jz{
    jz* son[2];
    int x,w,rd,s;
    int cmp(int k){if (k<x) return 0;else if (k==x) return -1;else return 1;}
    void updata(){s=son[0]->s+son[1]->s+w;}
};
jz a[maxn],*null=a,*ro=null;
int m,tot;
inline int _read(){
    int num=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') f=-f;ch=getchar();}
    while (ch>='0'&&ch<='9') num=num*10+ch-48,ch=getchar();
    return num*f;
}
jz* newnode(int x){
    a[++tot].w=a[tot].s=1;a[tot].x=x;a[tot].rd=rand();
    a[tot].son[0]=a[tot].son[1]=null;
    return &a[tot];
}
void turn(jz* &k,int p){
    jz* t=k->son[p];k->son[p]=t->son[p^1];t->son[p^1]=k;
    t->updata();k->updata();k=t;
}
void Insert(jz* &k,int x){
    if (k==null) {k=newnode(x);return;}
    int p=k->cmp(x);
    if (p==-1) k->w++;else{
        Insert(k->son[p],x);
        if (k->son[p]->rd>k->rd) turn(k,p);
    }
    k->updata();
}
void del(jz* &k,int x){
    if (k==null) return;
    int d=k->cmp(x);
    if (d==-1){
        if (k->w>1) k->w--;else
        if (k->son[0]==null) k=k->son[1];else
        if (k->son[1]==null) k=k->son[0];else{
            int son;if (k->son[0]->rd>k->son[1]->rd) son=0;else son=1;
            turn(k,son);if(k==null) return;del(k->son[son^1],x);
        }
        if (k==null) return;
    }else del(k->son[d],x);
    k->updata();
}
int ask1(jz* k,int x){
    if (k==null) return 0;
    int d=k->cmp(x);
    if (d==0) return ask1(k->son[0],x);else
    if (d==-1) return k->son[0]->s+1;else
    return k->son[0]->s+k->w+ask1(k->son[1],x);
}
int ask2(jz* &k,int y){
    if (k==null) return 0;
    if (y<=k->son[0]->s) return ask2(k->son[0],y);else
    if (y>k->son[0]->s+k->w) return ask2(k->son[1],y-k->son[0]->s-k->w);else return k->x;
}
int ask3(jz* &k,int y){
    if (k==null) return 0;
    int d=k->cmp(y);
    if (d==0){int x=ask3(k->son[0],y);if (x==0) return k->x;else return x;
    }else return ask3(k->son[1],y);
}
int ask4(jz* &k,int y){
    if (k==null) return 0;
    int d=k->cmp(y);
    if (d==1){int x=ask4(k->son[1],y);if (x==0) return k->x;else return x;
    }else return ask4(k->son[0],y);
}
int main(){
    freopen("exam.in","r",stdin);
    freopen("exam.out","w",stdout);
    m=_read();
    while (m--){
        int x=_read(),y=_read();
        if (x==1) Insert(ro,y);
        if (x==2) del(ro,y);
        if (x==3) printf("%d\n",ask1(ro,y));
        if (x==4) printf("%d\n",ask2(ro,y));
        if (x==5) printf("%d\n",ask4(ro,y));
        if (x==6) printf("%d\n",ask3(ro,y));
    }
    return 0;
}