P3369 【模板】普通平衡树

原题链接

题解1

stl模拟，注意lowerbound的效果和pos的返回值

code1

#include<bits/stdc++.h>
using namespace std;
vector<int> a;
int main()
{
    int n;
    cin>>n;
    while(n--)
    {
        int op,x;
        cin>>op>>x;
        if(op==1) a.insert(lower_bound(a.begin(),a.end(),x),x);
        else if(op==2) a.erase(lower_bound(a.begin(),a.end(),x));
        else if(op==3)
        {
            int pos=lower_bound(a.begin(),a.end(),x)-a.begin();//迭代器减迭代器
            cout<<pos+1<<endl;
        }
        else if(op==4) cout<<a[x-1]<<endl;
        else if(op==5)
        {
            int pos=lower_bound(a.begin(),a.end(),x)-a.begin();//pos返回的是大于等于x的第一个元素在序列里的下标
            if(pos>0) cout<<a[pos-1]<<endl;//前缀
        }
        else
        {
            int pos=lower_bound(a.begin(),a.end(),x+1)-a.begin();//大于等于x+1 相当于大于x
            if(a[pos]>x)  cout<<a[pos]<<endl;
        }
    }
    return 0;
}