【bzoj】3224: Tyvj 1728 普通平衡树

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 10097  Solved: 4302
[Submit][Status][Discuss]

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]

数据如下http://pan.baidu.com/s/1jHMJwO2

---

一万年没写数据结构了，来发treap模板练练手/

 #include<iostream>
 #include<cstdio>
 #include<algorithm>
 #include<vector>
 #include<cstdlib>
 #include<cmath>
 #include<cstring>
 using namespace std;
 #define maxn 110010
 #define llg  int
 #define yyj(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
 int n,m,ans;

 inline int getint()
 {
        int w=,q=;
        char c=getchar();
        while((c<'' || c>'') && c!='-') c=getchar();
        if (c=='-')  q=, c=getchar();
        while (c>='' && c<='') w=w*+c-'', c=getchar();
        return q ? -w : w;
 }

 struct MYTREAP
 {
     int root,size;
     struct
     {
         int l,r,rnd,val,w,size;
     }po[maxn];

     void init(){ memset(po,,sizeof(po)); root=size=;}

     void update(llg now) {po[now].size=po[po[now].l].size+po[po[now].r].size+po[now].w;}

     void right_(llg &now) {llg t=po[now].l; po[now].l=po[t].r; po[t].r=now; po[t].size=po[now].size; update(now); now=t;}
     void left_(llg &now)  {llg t=po[now].r; po[now].r=po[t].l; po[t].l=now; po[t].size=po[now].size; update(now); now=t;}

     void in(llg val)
     {
         insert(root,val);
     }

     void insert(llg &now,llg val)
     {
         if (now==)
         {
             now=++size;
             po[now].size=po[now].w=; po[now].val=val; po[now].rnd=rand();
             return ;
         }
         po[now].size++;
         if (po[now].val==val)
         {
             po[now].w++;
         }
         else
             if (po[now].val<val)
             {
                 insert(po[now].r,val);
                 if (po[po[now].r].rnd<po[now].rnd) left_(now);
             }
             else
             {
                 insert(po[now].l,val);
                 if (po[po[now].l].rnd<po[now].rnd) right_(now);
             }
     }

     void del(llg &now,llg val)
     {
         if (now==) return ;
         if (po[now].val==val)
         {
             if (po[now].w>)
             {
                 po[now].w--; po[now].size--;
                 return ;
             }
             if (po[now].l*po[now].r==) now=po[now].l+po[now].r;
             else
             {
                 if (po[po[now].l].rnd<po[po[now].r].rnd) right_(now); else left_(now);
                 del(now,val);
             }
         }
         else
         {
             po[now].size--;
             if (val>po[now].val) del(po[now].r,val);
             else del(po[now].l,val);
         }
     }

     int ask_rank(llg now,llg val)
     {
         if (now==) return ;
         if (po[now].val==val) return po[po[now].l].size+;
         else
             if (val>po[now].val) return po[po[now].l].size+po[now].w+ask_rank(po[now].r,val);
             else return ask_rank(po[now].l,val);
     }

     int ask_num(llg now,int val)
     {
         if (now==) return ;
         if (val<=po[po[now].l].size)
             return ask_num(po[now].l,val);
         else if (val>po[po[now].l].size+po[now].w)
             return ask_num(po[now].r,val-po[po[now].l].size-po[now].w);
         else
             return po[now].val;
     }

     void pre(llg now,llg val)
     {
         if (now==) return ;
         if (po[now].val<val)
         {
             ans=now;
             pre(po[now].r,val);
         }
         else pre(po[now].l,val);
     }

     void nex(llg now,llg val)
     {
         if (now==) return ;
         if (po[now].val>val)
         {
             ans=now;
             nex(po[now].l,val);
         }
         else nex(po[now].r,val);
     }

 }tree;

 int main()
 {
     yyj("");
     srand();
     tree.init();
     cin>>n;
     llg ty,x;
     for (llg i=;i<=n;i++)
     {
         ty=getint(),x=getint();
         switch(ty)
         {
             case :tree.in(x);break;
             case :tree.del(tree.root,x);break;
             case :printf("%d\n",tree.ask_rank(tree.root,x));break;
             case :printf("%d\n",tree.ask_num(tree.root,x));break;
             case :ans=;tree.pre(tree.root,x);printf("%d\n",tree.po[ans].val);break;
             case :ans=;tree.nex(tree.root,x);printf("%d\n",tree.po[ans].val);break;
         }
     }
     return ;
 }