treap板子（洛谷 P3369 【模板】普通平衡树（Treap/SBT））

由于有相同的数，每个节点加一个权值表示此数出现的次数

 #include<cstdio>
 #include<cstdlib>
 #include<ctime>
 #include<algorithm>
 using namespace std;
 #define MAXI 2147483647
 //http://blog.csdn.net/h348592532/article/details/52837228随机数
 int rand1()
 {
     static int x=;
     return x=(48271LL*x+)%;
 }
 struct Node
 {
     Node* ch[];
     int r;//优先级
     int v;//value
     int size;//维护子树的节点个数
     int num;//当前数字出现次数
     int cmp(int x) const//要在当前节点的哪个子树去查找,0左1右
     {
         if(x==v)    return -;
         return v<x;//x<v?0:1
     }
     void upd()
     {
         size=num;
         if(ch[]!=NULL)    size+=ch[]->size;
         if(ch[]!=NULL)    size+=ch[]->size;
     }
 }nodes[];
 int mem,n;
 Node* root=NULL;
 void rotate(Node* &o,int d)
 {
     Node* t=o->ch[d^];o->ch[d^]=t->ch[d];t->ch[d]=o;
     o->upd();t->upd();//o是t子节点，一定要这个顺序upd
     o=t;//将当前节点变成旋转完后新的父节点
 }
 Node* getnode()
 {
     return &nodes[mem++];
 }
 void insert(Node* &o,int x)
 {
     if(o==NULL)
     {
         o=getnode();o->ch[]=o->ch[]=NULL;
         o->v=x;o->r=rand1();o->num=;
     }
     else
     {
         if(o->v==x)    ++(o->num);
         else
         {
             int d=o->v < x;//x < o->v?0:1
             insert(o->ch[d],x);
             if(o->r < o->ch[d]->r)    rotate(o,d^);//不是 x < o->ch[d]->r
         }
     }
     o->upd();
 }
 void remove(Node* &o,int x)
 {
     int d=o->cmp(x);
     if(d==-)
     {
         if(o->num > )
         {
             --(o->num);
         }
         if(o->num == )
         {
             if(o->ch[]==NULL)    o=o->ch[];
             else if(o->ch[]==NULL)    o=o->ch[];
             else
             {
                 //int d2= o->ch[0]->r > o->ch[1]->r;//o->ch[0]->r > o->ch[1]->r ? 1:0
                 int d2=o->ch[]->r < o->ch[]->r;//o->ch[1]->r <= o->ch[0]->r
                 rotate(o,d2);
                 remove(o->ch[d2],x);
                 //左旋则原节点变为新节点的左子节点，右旋相反
             }
         }
     }
     else    remove(o->ch[d],x);
     if(o!=NULL)    o->upd();
 }
 bool find(Node* o,int x)
 {
     int d;
     while(o!=NULL)
     {
         d=o->cmp(x);
         if(d==-)    return ;
         else o=o->ch[d];
     }
     return ;
 }
 int kth(Node* o,int k)
 {
     if(o==NULL||k<=||k > o->size)    return ;
     int s= o->ch[]==NULL ?  : o->ch[]->size;
     if(k>s&&k<=s+ o->num)    return o->v;
     else if(k<=s)    return kth(o->ch[],k);
     else    return kth(o->ch[],k-s- o->num);
 }
 int rk(Node* o,int x)
 {
     int r=o->ch[]==NULL ?  : o->ch[]->size;
     if(x==o->v)    return r+;
     else    if(x<o->v)    return rk(o->ch[],x);
     else    return r+ o->num +rk(o->ch[],x);
 }
 int pre(Node* o,int x)
 {
     if(o==NULL)    return -MAXI;
     int d=o->cmp(x);
     if(d<=)    return pre(o->ch[],x);
     else    return max(o->v,pre(o->ch[],x));
 }
 int nxt(Node* o,int x)
 {
     if(o==NULL)    return MAXI;
     int d=o->cmp(x);
     if(d!=)    return nxt(o->ch[],x);
     else    return min(o->v,nxt(o->ch[],x));
 }
 int main()
 {
     int i,o,x;
     scanf("%d",&n);
     for(i=;i<=n;i++)
     {
         scanf("%d%d",&o,&x);
         if(o==)    insert(root,x);
         else if(o==)    remove(root,x);
         else if(o==)    printf("%d\n",rk(root,x));
         else if(o==)    printf("%d\n",kth(root,x));
         else if(o==)    printf("%d\n",pre(root,x));
         else printf("%d\n",nxt(root,x));
     }
     return ;
 }

本来想写成相同的数分成多个节点放的，写了一会儿各种细节，炸了

 #include<cstdio>
 #include<cstdlib>
 #include<ctime>
 #include<algorithm>
 using namespace std;
 struct Node
 {
     Node* ch[];
     int r;//优先级
     int v;//value
     int size;//维护子树的节点个数

     int cmp(int x) const//要在当前节点的哪个子树去查找,0左1右
     {
         if(x==v)    return -;
         return v<x;//x<v?0:1
     }
     void upd()
     {
         size=;
         if(ch[]!=NULL)    size+=ch[]->size;
         if(ch[]!=NULL)    size+=ch[]->size;
     }
 }nodes[];
 int mem;
 void rotate(Node* &o,int d)
 {
     Node* t=o->ch[d^];o->ch[d^]=t->ch[d];t->ch[d]=o;
     o->upd();t->upd();
     o=t;//将当前节点变成旋转完后新的父节点
 }
 Node* getnode()
 {
     return &nodes[mem++];
 }
 void insert(Node* &o,int x)
 {
     if(o==NULL)
     {
         o=getnode();o->ch[]=o->ch[]=NULL;
         o->v=x;o->r=rand();
     }
     else
     {
         //int d=o->cmp(x);不重复
         int d=o->v <= x;//x < o->v?0:1
         //按照以上左侧的插入方式，相等的会在左子树中插入
         insert(o->ch[d],x);
         if(o->ch[d]->r > o->r)//不是> x
             rotate(o,d^);
     }
     o->upd();
 }
 void remove(Node* &o,int x)
 {
     int d=o->cmp(x);
     if(d==-)
     {
         if(o->ch[]==NULL)    o=o->ch[];
         else if(o->ch[]==NULL)    o=o->ch[];
         else
         {
             //int d2= o->ch[0]->r > o->ch[1]->r;//o->ch[0]->r > o->ch[1]->r ? 1:0
             int d2=o->ch[]->r <= o->ch[]->r;
             rotate(o,d2);
             remove(o->ch[d2],x);
             //左旋则原节点变为新节点的左子节点，右旋相反
         }
     }
     else    remove(o->ch[d],x);
     if(o!=NULL)    o->upd();
 }
 bool find(Node* o,int x)
 {
     int d;
     while(o!=NULL)
     {
         d=o->cmp(x);
         if(d==-)    return ;
         else o=o->ch[d];
     }
     return ;
 }
 int kth(Node* o,int k)
 {
     if(o==NULL||k<=||k > o->size)    return ;
     int s= o->ch[]==NULL ?  : o->ch[]->size;
     if(k==s+)    return o->v;
     else if(k<=s)    return kth(o->ch[],k);
     else    return kth(o->ch[],k-s-);
 }
 //int rank(Node* o,int x)//不可重复
 //{
 //    int r= t->ch[0]==NULL ? 0 : t->ch[0]->size;
 //    if(x==o->v)    return r+1;
 //    else    if(x<o->v)    return rank(o->ch[0],x);
 //    else    return r+1+rank(o->ch[1],x);
 //}
 int rank(Node* o,int x)//支持重复
 {
     int r=t->ch[]==NULL ?  : t->ch[]->size;
     if(x==o->v)
     {
         int t=rank(o->ch[
     }
 }
 int pre(Node* o,int x)
 {
     if(o==NULL)    return -;
     if(x==o->v)    return
     if(x<o->v)    return min(o->v,pre(o->ch[],
 }
 int main()
 {
     srand(time());

     return ;
 }