洛谷P3369 【模板】普通平衡树

题目描述

您需要写一种数据结构（可参考题目标题），来维护一些数，其中需要提供以下操作：

1. 插入x数
2. 删除x数(若有多个相同的数，因只删除一个)
3. 查询x数的排名(排名定义为比当前数小的数的个数+1。若有多个相同的数，因输出最小的排名)
4. 查询排名为x的数
5. 求x的前驱(前驱定义为小于x，且最大的数)
6. 求x的后继(后继定义为大于x，且最小的数)

输入输出格式

输入格式：

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

输出格式：

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

输入输出样例

输入样例#1： 

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

输出样例#1： 

106465
84185
492737

说明

时空限制：1000ms,128M

1.n的数据范围： n≤100000

2.每个数的数据范围： [-10^7, 10^7]

来源：Tyvj1728 原名：普通平衡树

在此鸣谢

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
long long read()
{
    long long x=0,f=1;
    char ch=getchar();
    while(ch>'9'||ch<'0')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
    {
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
vector<int>v;
int n,opt,x;
int main()
{
    v.reserve(100001);
    n=read();
    while(n--)
    {
        opt=read(),x=read();
        if(opt==1)
            v.insert(lower_bound(v.begin(),v.end(),x),x);
        if(opt==2)
            v.erase (lower_bound(v.begin(),v.end(),x));
        if(opt==3)
            printf("%d\n",lower_bound(v.begin(),v.end(),x)-v.begin()+1);
        if(opt==4)
            printf("%d\n",v[x-1]);
        if(opt==5)
            printf("%d\n",v[lower_bound(v.begin(),v.end(),x)-v.begin()-1]);
        if(opt==6)
            printf("%d\n",v[upper_bound(v.begin(),v.end(),x)-v.begin()]);
    }
    return 0;
}