bzoj 3224: Tyvj 1728 普通平衡树 Treap模版

3224: Tyvj 1728 普通平衡树

Time Limit: 10 Sec  Memory Limit: 128 MB
Submit: 16072  Solved: 6973
[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.每个数的数据范围：[-2e9,2e9]

 

Source

平衡树

----------------------------------------------------------------------------------------------------------------------------

感谢brian551神犇提供的模版orz

啃了一星期的平衡树，总算是啃下来了。

二叉搜索树+小/大根堆=Treap 

讲得很详细：http://blog.csdn.net/u014634338/article/details/49612159

 

  1 #include<stdio.h>
  2 #include<string.h>
  3 #include<stdlib.h>
  4 #define maxn 233333
  5 struct node{
  6     int l,r,v,w,size,rnd;
  7 };
  8 node tr[maxn*2];
  9 int read();
 10 void update(int);
 11 void lturn(int&);
 12 void rturn(int&);
 13 void insert(int&,int);
 14 void del(int&,int);
 15 int query_rank(int,int);
 16 int query_num(int,int);
 17 void query_pro(int,int);
 18 void query_sub(int,int);
 19 int n,size,root,an;
 20 int main(){
 21     n=read();size=0;root=0;an=0;
 22     for(int i=1;i<=n;i++){
 23         int opt=read(),x=read();
 24         switch(opt){
 25             case 1: insert(root,x);break;
 26             case 2: del(root,x);break;
 27             case 3: printf("%d\n",query_rank(root,x));break;
 28             case 4: printf("%d\n",query_num(root,x));break;
 29             case 5: an=0;query_pro(root,x);printf("%d\n",tr[an].v);break;
 30             case 6: an=0;query_sub(root,x);printf("%d\n",tr[an].v);break;
 31         }
 32     }
 33     return 0;
 34 }
 35 void update(int k){
 36     tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;
 37 }
 38 void lturn(int &k){
 39     int t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;
 40     tr[t].size=tr[k].size;update(k);k=t;
 41 }
 42 void rturn(int &k){
 43     int t=tr[k].l;tr[k].l=tr[t].r;tr[t].r=k;
 44     tr[t].size=tr[k].size;update(k);k=t;
 45 }
 46 void insert(int &k,int x){
 47     if(k==0){
 48         k=++size;
 49         tr[k].w=tr[k].size=1;tr[k].rnd=rand();tr[k].v=x;
 50         return;
 51     }
 52     tr[k].size++;
 53     if(tr[k].v==x) tr[k].w++;
 54     else if(tr[k].v>x){
 55         insert(tr[k].l,x);
 56         if(tr[k].rnd>tr[tr[k].l].rnd) rturn(k);
 57     }
 58     else{
 59         insert(tr[k].r,x);
 60         if(tr[k].rnd>tr[tr[k].r].rnd) lturn(k);
 61     }
 62 }
 63 void del(int &k,int x){
 64     if(k==0) return;
 65     if(tr[k].v==x){
 66         if(tr[k].w>1){
 67             tr[k].w--;tr[k].size--;return;
 68         }
 69         if(tr[k].r*tr[k].l==0) k=tr[k].l+tr[k].r;
 70         else if(tr[tr[k].l].rnd>tr[tr[k].r].rnd) lturn(k),del(k,x);
 71         else rturn(k),del(k,x);
 72     }
 73     else if(tr[k].v>x) tr[k].size--,del(tr[k].l,x);
 74     else tr[k].size--,del(tr[k].r,x);
 75 }
 76 int query_rank(int k,int x){
 77     if(k==0) return 0;
 78     if(tr[k].v==x) return tr[tr[k].l].size+1;
 79     else if(tr[k].v<x) return tr[tr[k].l].size+tr[k].w+query_rank(tr[k].r,x);
 80     else return query_rank(tr[k].l,x);
 81 }
 82 int query_num(int k,int x){
 83     if(k==0) return 0;
 84     if(x<=tr[tr[k].l].size) return query_num(tr[k].l,x);
 85     else if(x>tr[tr[k].l].size+tr[k].w) return query_num(tr[k].r,x-tr[tr[k].l].size-tr[k].w);
 86     else return tr[k].v;
 87 }
 88 void query_pro(int k,int x){
 89     if(k==0) return;
 90     if(x>tr[k].v) an=k,query_pro(tr[k].r,x);
 91     else query_pro(tr[k].l,x);
 92 }
 93 void query_sub(int k,int x){
 94     if(k==0) return;
 95     if(x<tr[k].v) an=k,query_sub(tr[k].l,x);
 96     else query_sub(tr[k].r,x);
 97 }
 98 int read(){
 99     int ans=0,f=1;char c=getchar();
100     while('0'>c||c>'9'){if(c=='-')f=-1;c=getchar();}
101     while('0'<=c&&c<='9')ans=ans*10+c-48,c=getchar();return ans*f;
102 }
Treap

 

posted @ 2017-09-26 21:56  lpl_bys  阅读(155)  评论(0)  编辑  收藏  举报