题解 | 【模板】集合操作

#include<bits/stdc++.h>
using namespace std;
set <int> s;//元素唯一，且从小到大排列，multiset则可重复，基于红黑树
void insertValue(int x){
    //TODO 实现插入逻辑
    s.insert(x);
}
void eraseValue(int x){
    //TODO 实现删除逻辑']
    s.erase(x);
}
int xInSet(int x){
    //TODO 实现存在性检查
    // if (s.count(x) == 0) return false;
    // return true;
    return s.count(x);
}
int sizeOfSet(){
    //TODO 返回集合大小
    return s.size();
}
int getPre(int x){
    //TODO 实现找前驱
    auto it = s.lower_bound(x);
    if (it == s.begin()) return -1;
    it --;
    return *it;
}
int getBack(int x){
    //TODO 实现找后继
    auto it  = s.upper_bound(x);
    if  (it == s.end()) return -1;
    return *it;//不用再it++，因为upper_bound返回第一个大于target的元素，lower_bound放回第一个大于等于target的元素，所以迭代器需要好指向更小一点的元素
}
int main(){
    int q,op,x;
    cin>>q;
    while(q--){
        cin>>op;
        if(op==1){
            cin>>x;
            insertValue(x);
        }
        if(op==2){
            cin>>x;
            eraseValue(x);
        }
        if(op==3){
            cin>>x;
            if(xInSet(x)){
                cout<<"YES\n";
            }else{
                cout<<"NO\n";
            }
        }
        if(op==4){
            cout<<sizeOfSet()<<endl;
        }
        if(op==5){
            cin>>x;
            cout<<getPre(x)<<endl;
        }
        if(op==6){
            cin>>x;
            cout<<getBack(x)<<endl;
        }
    }
    return 0;
}