题解 | 【模板】集合操作

#include<bits/stdc++.h>
using namespace std;set<int>s;
int getPre(int x){
    auto it=s.lower_bound(x);
    return it==s.begin()?-1:*prev(it);
}
int getBack(int x){
    auto it=s.upper_bound(x);
    return it==s.end()?-1:*it;
}
int main(){
    int q,op,x;cin>>q;
    while(q--){cin>>op;
        if(op-4)cin>>x;
        if(op==1)s.insert(x);
        if(op==2)s.erase(x);
        if(op==3)cout<<(s.count(x)?"YES":"NO");
        if(op==4)cout<<s.size();
        if(op==5)cout<<getPre(x);
        if(op==6)cout<<getBack(x);
        if(op>2)puts("");
    }
    return 0;
}

考察set的基本使用语句