牛客小白月赛6 - G - 指纹锁（set二分）

牛客小白月赛6 - G - 指纹锁

#include <iostream>
#include <string.h>
#include <stdio.h>
#include <algorithm>
#include <set>
using namespace std;
typedef long long int LL;
set<int>st;
set<int>::iterator it;
int m, k, x;
char s[10];
bool query(int x){
    it = st.lower_bound(x);
    if(!st.empty() && it!=st.end() && *it <= x+k*2) return true;
    return false;
}
void del(int x){
    it = st.lower_bound(x);
    while(!st.empty() && it!=st.end() && *it <= x+k*2) {
        int val = *it; it ++;
        st.erase(val);
    }
}
int main()
{
    scanf("%d%d", &m, &k);
    for(int i=1;i<=m;i++){
        scanf("%s%d", s, &x);
        if(s[0] == 'a') {
            if(!query(x-k)) st.insert(x);
        }
        else if(s[0] == 'd') del(x-k);
        else printf("%s\n",query(x-k)?"Yes":"No");
    }
    return 0;
}