牛客小白月赛8 F.数列操作

牛客小白月赛8 F.数列操作

题目链接

题目描述

clccle是个蒟蒻,她经常会在学校机房里刷题,也会被同校的dalao们虐,有一次,她想出了一个毒瘤数据结构,便兴冲冲的把题面打了出来,她觉得自己能5s内切掉就很棒了,结果evildoer过来一看,说:“这思博题不是1s就能切掉嘛”,clccle觉得自己的信心得到了打击,你能帮她在1s中切掉这道水题嘛?

你需要写一个毒瘤(划掉)简单的数据结构,满足以下操作
1.插入一个数x(insert)
2.删除一个数x(delete)(如果有多个相同的数,则只删除一个)
3.查询一个数x的排名(若有多个相同的数，就输出最小的排名)
4.查询排名为x的数
5.求一个数x的前驱
6.求一个数x的后继

输入描述:

第一行,输入一个整数n,表示接下来需要输入n行

接下来n行,输入 一个整数num和一个整数x

输出描述:

当num为3,4,5,6时,输出对应的答案

示例1

输入

8
1 10
1 20
1 30
3 20
4 2
2 10
5 25
6 -1

输出

2
20
20
20

题解要用 splay 树，貌似直接 vector 就可以搞定，简单数据结构，AC代码如下：

#include 
using namespace std;
typedef long long ll;
int n,op,x;
vector<int>v;
int main(){
     
    scanf("%d",&n);
    while(n--){
     
        scanf("%d%d",&op,&x);
        if(op==1){
     
            v.insert(upper_bound(v.begin(),v.end(),x),x);
        }else if(op==2){
     
            v.erase(lower_bound(v.begin(),v.end(),x));
        }else if(op==3){
     
            printf("%d\n",lower_bound(v.begin(),v.end(),x)-v.begin()+1);
        }else if(op==4){
     
            printf("%d\n",v[x-1]);
        }else if(op==5){
     
            printf("%d\n",*--lower_bound(v.begin(),v.end(),x));
        }else{
     
            printf("%d\n",*upper_bound(v.begin(),v.end(),x));
        }
    }
}