普通平衡树Tyvj1728、luogu P3369 （treap）

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

1. 插入x数

2. 删除x数(若有多个相同的数，因只删除一个)

3. 查询x数的排名(若有多个相同的数，因输出最小的排名)

4. 查询排名为x的数

5. 求x的前驱(前驱定义为小于x，且最大的数)

6. 求x的后继(后继定义为大于x，且最小的数)

             --by 洛谷；

http://www.lydsy.com/JudgeOnline/problem.php?id=3224

---

---

　　话说，没时间写题解啊——

　　不过，也就是个平衡树的模板题；

    就存下代码吧

　　treap哟；

　　好像很少见写treap的，但常数还是挺小的orz；

　　哎，最近怎么老写模板啊？

　　代码如下：

#include<cstdio>
#include<cstdlib>
using namespace std;
#define INF 2147483647
int n;
struct poo
{
    int size,value,key,cnt;
    int ch[2];
}data[100001];
int tot,root,x;
int  make_data(int );
void insert(int&);
void roll(int&);
int  find(    );
int  rank(    );
void del(int&);
int  las(int );
int  nex(int );
void up(int );
int main()
{
    int i,j;
    data[0].value=INF;data[0].key=INF;
    scanf("%d",&n);
    for(i=1;i<=n;i++){
        scanf("%d%d",&j,&x);
        switch(j){
            case 1:insert(root);break;
            case 2:   del(root);break;
            case 3:  printf("%d
",rank(    ));break;
            case 4:  printf("%d
",find(    ));break;
            case 5:  printf("%d
", las(root));break;
            case 6:  printf("%d
", nex(root));break;
        }
    }
}
int make_data(int value)
{
    tot++;
    data[tot].cnt++;
    data[tot].key=(rand()/2+rand()/2);
    data[tot].size=1;
    data[tot].value=value;
    return tot;
}
void insert(int &now)
{
    if(now==0){
        now=make_data(x);
        return;
    }
    if(data[now].value==x){
        data[now].cnt++;
        data[now].size++;
    }
    else{
        int wh=x < data[now].value ? 0 : 1;
        insert(data[now].ch[wh]);
        if(data[now].key>=data[data[now].ch[wh]].key)
            roll(now);
    }
    up(now);
}
void roll(int &now)
{
    int wh=data[data[now].ch[0]].key<data[data[now].ch[1]].key?0:1;
    int son=data[now].ch[wh];
    data[now].ch[wh]=data[son].ch[wh^1];
    data[son].ch[wh^1]=now;
    up(now);
    now=son;
}
int find()
{
    int now=root;
    int ls,rs;
    ls=data[now].ch[0];rs=data[now].ch[1];
    while(x<=data[ls].size||x>data[now].size-data[rs].size){
        if(data[ls].size>=x)
            now=ls;
        else{
            x=x+data[rs].size-data[now].size;
            now=rs;
        }
        ls=data[now].ch[0];rs=data[now].ch[1];
    }
    return data[now].value;
}
int rank()
{
    int now=root,ans=0;
    int ls=data[now].ch[0],rs=data[now].ch[1];
    while(x!=data[now].value&&x!=0)
    {
        if(x<data[now].value)
            now=ls;
        else{
            ans+=data[now].size-data[rs].size;
            now=rs;
        }
        ls=data[now].ch[0];rs=data[now].ch[1];
    }
    return ans+data[ls].size+1;
}
void del(int &now)
{
    if(data[now].value==x){
        if(data[now].cnt==1){
            if(data[now].ch[0]*data[now].ch[1]==0){
                now=data[now].ch[1]+data[now].ch[0];
                return ;
            }
            roll(now);
            int wh=data[data[now].ch[0]].value==x?0:1;
            del(data[now].ch[wh]);
        }
        else{
            data[now].size--;    data[now].cnt--;
        }
    }
    else{
        int wh=data[now].value>x?0:1;
        del(data[now].ch[wh]);
    }
    up(now);
}
int las(int now)
{
    int ans=0,an=0;
    if(!now)return 0;
    if(data[now].value<x){
        ans=data[now].value;
        an=las(data[now].ch[1]);
        ans=an!=0?an:ans;
    }
    else{
        ans=las(data[now].ch[0]);
    }
    return ans;
}
int nex(int now)
{
    int ans=0,an=0;
    if(!now)return 0;
    if(data[now].value>x){
        ans=data[now].value;
        an=nex(data[now].ch[0]);
        ans=an!=0?an:ans;
    }
    else{
        ans=nex(data[now].ch[1]);
    }
    return ans;
}
void up(int now)
{
    data[now].size=data[data[now].ch[0]].size+data[data[now].ch[1]].size+data[now].cnt;
}
//treap on the 2017.1.21
//10
//1 5
//4 1
//1 6
//1 7
//1 10
//1 3
//1 4
//6 2
//1 8
//5 9
//
//14
//1 5
//1 6
//1 7
//1 10
//1 3
//1 4
//1 8
//3 3
//3 4
//3 5
//3 6
//4 5
//4 6
//4 7

祝AC哟；