BZOJ 3224 TREAP

【题目描述】
您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)
【解题报告】
这道题是一个比较全面的平衡树的题。他的操作比较多样。代码如下：

#include <iostream>#include <cstdio>#include <cstdlib>#include <cstring>using namespace std;#define INF 1000000000struct node{    int v;    int s;    int sz;    node *p;    node *ch[2];    node(){        v=sz=0;        s=-1;        p=ch[0]=ch[1]=this;    }    bool cmp(int s)    {        return this->s<s;    }    int cd(node *o)    {        return this->ch[1]==o?1:0;    }}Tnull,*null=&Tnull;void newnode(node *pre,int v,int d){    node *u=(node*)malloc(sizeof(node));    u->ch[0]=u->ch[1]=null;    u->sz=1;    u->s=rand();    u->v=v;    u->p=pre;    pre->ch[d]=u;}void pushup(node *o){    o->sz=o->ch[0]->sz+o->ch[1]->sz+1;}void rot(node *o,int d){    node *k=o->ch[d];    o->ch[d]=k->ch[d^1];    k->ch[d^1]->p=o;    o->p->ch[o->p->cd(o)]=k;    k->p=o->p;    k->ch[d^1]=o;    o->p=k;    pushup(o);    pushup(k);}void Insert(node *o,node *pre,int v,int dd){    if(o==null){        newnode(pre,v,dd);        return;    }    int d;    if(v>=o->v){        d=1;        Insert(o->ch[1],o,v,1);    }    else {        d=0;        Insert(o->ch[0],o,v,0);    }    if(o->cmp(o->ch[d]->s))rot(o,d);    else pushup(o);}void Delete(node *o,int v){    if(o==null)return;    if(o->v==v){        while(o->ch[0]!=null&&o->ch[1]!=null)        {            int d=o->ch[0]->cmp(o->ch[1]->s);            rot(o,d);            --o->p->sz;        }        node *k;        o->ch[0]==null ? k=o->ch[1] : k=o->ch[0];        o->p->ch[o->p->cd(o)]=k;        k->p=o->p;        free(o);        return;    }    v>o->v? Delete(o->ch[1],v) : Delete(o->ch[0],v);    pushup(o);}int Find(node *o,int v){    if(o==null)return 1;    int ans=0;    if(v<=o->v)ans=Find(o->ch[0],v);    else ans=Find(o->ch[1],v)+o->ch[0]->sz+1;    return ans;}int Find2(node *o,int k){    if(o==null)return -1;    if(o->ch[0]->sz+1==k)return o->v;    return o->ch[0]->sz>=k? Find2(o->ch[0],k) : Find2(o->ch[1],k-1-o->ch[0]->sz) ;}int mini;int maxi;void Findp(node *o,int v){    if(o==null)return;    if(o->v<v){        if(maxi<o->v)maxi=o->v;        Findp(o->ch[1],v);    }    else Findp(o->ch[0],v);}void Findh(node *o,int v){    if(o==null)return;    if(o->v>v){        if(mini>o->v)mini=o->v;        Findh(o->ch[0],v);    }    else Findh(o->ch[1],v);}void Free(node *o){    if(o==null)return;    Free(o->ch[0]);    Free(o->ch[1]);    free(o);}int main(){    int n;    while(scanf("%d",&n)!=EOF)    {        node *root=null;        for(int i=0;i<n;++i)        {            int vis,v;            scanf("%d%d",&vis,&v);            if(vis==1)Insert(root->ch[1],root,v,1);            else if(vis==2)Delete(root->ch[1],v);            else if(vis==3)printf("%d\n",Find(root->ch[1],v));            else if(vis==4)printf("%d\n",Find2(root->ch[1],v));            else if(vis==5)            {                maxi=-INF;                Findp(root->ch[1],v);                printf("%d\n",maxi);            }            else            {                mini=INF;                Findh(root->ch[1],v);                printf("%d\n",mini);            }        }        Free(root->ch[1]);        root->ch[1]=null;    }    return 0;}