3224: Tyvj 1728 普通平衡树

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]

 

模板题

 #include<cstdio>
 #include<cstdlib>
 using namespace std;
 struct tree{
     int l,r,sum,s,num,rnd;
 }tr[];
 int cnt,t1,t2,n,opt,x,root;
 void updata(int k){
     int l=tr[k].l,r=tr[k].r;
     tr[k].s=tr[l].s+tr[r].s+tr[k].sum;
 }

 void lturn(int &k){
     int t=tr[k].r;tr[k].r=tr[t].l;tr[t].l=k;
     tr[t].s=tr[k].s;updata(k);k=t;
 }

 void rturn(int &k){
     int t=tr[k].l; tr[k].l=tr[t].r;tr[t].r=k;
     tr[t].s=tr[k].s;updata(k);k=t;
 }

 void insert(int &k,int x){
 if (!k){
         cnt++;k=cnt;tr[k].num=x; tr[k].s=; tr[k].rnd=rand();tr[k].sum++;return;
     }
     tr[k].s++;
     if (x<tr[k].num){
         insert(tr[k].l,x);
         if (tr[tr[k].l].rnd<tr[k].rnd) rturn(k);
     }
     else if (x>tr[k].num){
         insert(tr[k].r,x);
         if (tr[tr[k].r].rnd<tr[k].rnd) lturn(k);
     }
     else if (x==tr[k].num) {
         tr[k].sum++;
         return;
     }
 }

 void del(int &k,int x){
     if (k==) return;
     if (tr[k].num==x){
         if (tr[k].sum>){
             tr[k].sum--;tr[k].s--; 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].num) tr[k].s--,del(tr[k].r,x);
     else tr[k].s--,del(tr[k].l,x);
 }

 int find1(int &k,int x){
     if (k==) return ;
     if (tr[k].num==x) return tr[tr[k].l].s+;
     if (tr[k].num>x) return find1(tr[k].l,x);
     if (tr[k].num<x) return tr[tr[k].l].s+tr[k].sum+find1(tr[k].r,x);
 } 

 int find2(int &k,int x){
     if (k==) return ;
     int l=tr[k].l,r=tr[k].r;
     if (tr[l].s+<=x&&tr[l].s+tr[k].sum>=x) return tr[k].num;
     if (tr[l].s>=x) return find2(l,x);
     if (tr[l].s+tr[k].sum<x) return find2(r,x-tr[l].s-tr[k].sum);
 }

 void pre(int &k,int x){
     if (k==) return;
     if (tr[k].num<x){
         t1=k;
         pre(tr[k].r,x);
     }
     else pre(tr[k].l,x);
 }

 void after(int &k,int x){
     if (k==) return;
     if (tr[k].num>x){
         t2=k;
         after(tr[k].l,x);
     }
     else after(tr[k].r,x);
 }

 int main(){
     scanf("%d",&n);
     while (n--){
         scanf("%d%d",&opt,&x);
         t1=t2=;
         switch(opt){
             case :insert(root,x); break;
             case :del(root,x); break;
             case :printf("%d\n",find1(root,x)); break;//查询x数的排名
             case :printf("%d\n",find2(root,x));break;
             case :pre(root,x);printf("%d\n",tr[t1].num);break;
             case :after(root,x);printf("%d\n",tr[t2].num);break;
         }

     }
 }