bzoj3224（treap）

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec   Memory Limit: 128 MB
Submit: 1844   Solved: 727
[ Submit][ Status]

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每行输出一个数，表示对应答案

Sample Input

10
1 106465
4 1
1 317721
1 460929
1 644985
1 84185
1 89851
6 81968
1 492737
5 493598

Sample Output

106465
84185
492737

HINT

1.n的数据范围：n<=100000

2.每个数的数据范围：[-1e7,1e7]

题意：RT

思路：平衡树裸题，我用treap实现的，由于敲rot的时候少敲了一个字符导致无限RE，sad~,还是得小心的敲哎，跑完412ms

            这题有个坑，对于操作3，要找出最小排名，比如现在有1 3 3 5   那么在输入3 3以后就要输出2才对，其它的也没啥要注意的了，放开写就行了~

#include 
   
   
    
    
#include 
    
    
     
     
#include 
     
     
      
      
#include 
      
      
       
       
using namespace std;
#define INF 1000000000

struct 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
       
       
         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)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 
         
           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; }