c++之路进阶——codevs4543（普通平衡树）

4543 普通平衡树

 时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 大师 Master

 

 

 

题目描述 Description

这是一道水题

顺便祝愿ＬＥＺ和ＺＱＱ　省选ＡＫＡＫＡＫ

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：
1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

输入描述 Input Description

第一行为n，表示操作的个数,下面n行每行有两个数opt和x，opt表示操作的序号(1<=opt<=6)

输出描述 Output Description

对于操作3,4,5,6每行输出一个数，表示对应答案

样例输入 Sample Input

10

1 10

1 10

1 10

1 10

1 10

1 10

1 10

1 10

1 10

1 10

样例输出 Sample Output

EOF（无输出）

数据范围及提示 Data Size & Hint

n=100000 所有数字均在-2*10^9到2*10^9内

其实n=5000000才对。。。但是为了不卡评测机

题解：

   没有任何技术含量，就是treap基本操作。

代码：

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 using namespace std;
 struct data{
     int l,r,v,size,rnd,w;
 }tr[];
 int n,size,root,ans;
 void update(int k)//更新结点信息
 {
     tr[k].size=tr[tr[k].l].size+tr[tr[k].r].size+tr[k].w;
 }
 void rturn(int &k)
 {
     int t=tr[k].l;tr[k].l=tr[t].r;tr[t].r=k;
     tr[t].size=tr[k].size;update(k);k=t;
 }
 void lturn(int &k)
 {
     int t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;
     tr[t].size=tr[k].size;update(k);k=t;
 }
 void insert(int &k,int x)
 {
     if(k==)
     {
     size++;k=size;
     tr[k].size=tr[k].w=;tr[k].v=x;tr[k].rnd=rand();
     return;
     }
     tr[k].size++;
     if(tr[k].v==x)tr[k].w++;//每个结点顺便记录下与该节点相同值的数的个数
     else if(x>tr[k].v)
     {
     insert(tr[k].r,x);
     if(tr[tr[k].r].rnd<tr[k].rnd)lturn(k);//维护堆性质
     }
     else
     {
     insert(tr[k].l,x);
     if(tr[tr[k].l].rnd<tr[k].rnd)rturn(k);
     }
 }
 void del(int &k,int x)
 {
     if(k==)return;
     if(tr[k].v==x)
     {
     if(tr[k].w>)
     {
     tr[k].w--;tr[k].size--;return;//若不止相同值的个数有多个，删去一个
     }
     if(tr[k].l*tr[k].r==)k=tr[k].l+tr[k].r;//有一个儿子为空
     else if(tr[tr[k].l].rnd<tr[tr[k].r].rnd)
     rturn(k),del(k,x);
     else lturn(k),del(k,x);
     }
     else if(x>tr[k].v)
     tr[k].size--,del(tr[k].r,x);
     else tr[k].size--,del(tr[k].l,x);
 }
 int query_rank(int k,int x)
 {
     if(k==)return ;
     if(tr[k].v==x)return tr[tr[k].l].size+;
     if(x>tr[k].v)return tr[tr[k].l].size+tr[k].w+query_rank(tr[k].r,x);
     else return query_rank(tr[k].l,x);
 }
 int query_num(int k,int x)
 {
     if(k==)return ;
     if(x<=tr[tr[k].l].size)
     return query_num(tr[k].l,x);
     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);
     else return tr[k].v;
 }
 void query_pro(int k,int x)
 {
     if(k==)return;
     if(tr[k].v<x)
     {
     ans=k;query_pro(tr[k].r,x);
     }
     else query_pro(tr[k].l,x);
 }
 void query_sub(int k,int x)
 {
     if(k==)return;
     if(tr[k].v>x)
     {
     ans=k;query_sub(tr[k].l,x);
     }
     else query_sub(tr[k].r,x);
 }
 int main()
 {
     scanf("%d",&n);
     int opt,x;
     for(int i=;i<=n;i++)
     {
     scanf("%d%d",&opt,&x);
     switch(opt)
     {
     case :insert(root,x);break;
     case :del(root,x);break;
     case :printf("%d\n",query_rank(root,x));break;
     case :printf("%d\n",query_num(root,x));break;
     case :ans=;query_pro(root,x);printf("%d\n",tr[ans].v);break;
     case :ans=;query_sub(root,x);printf("%d\n",tr[ans].v);break;
     }
     }
     return ;
 }