STL之set重载()

题目描述

HA实验有一套非常严密的安全保障体系，在HA实验基地的大门，有一个指纹锁。

该指纹锁的加密算***把一个指纹转化为一个不超过1e7的数字，两个指纹数值之差越小，就说明两个指纹越相似，当两个指纹的数值差≤k时，这两个指纹的持有者会被系统判定为同一个人。
现在有3种操作，共m个，
操作1：add x，表示为指纹锁录入一个指纹，该指纹对应的数字为x，如果系统内有一个与x相差≤k的指纹，则系统会忽略这次添加操作
操作2：del x，表示删除指纹锁中的指纹x，若指纹锁中多个与x相差≤k的指纹，则全部删除，若指纹锁中没有指纹x，则可以忽略该操作，
操作3：query x，表示有一个持有指纹x的人试图打开指纹锁，你需要设计一个判断程序，返回该人是否可以打开指纹锁（只要x与存入的任何一个指纹相差≤k即可打开锁）。
初始状态，指纹锁中没有任何指纹。

输入描述:

第一行有2个正整数m，k。
接下来m行，每行描述一种操作：add x，del x或query x。

输出描述:

对于每个query操作，输出一行，包含一个单词“Yes”或“No”，表示该人是否可以打开指纹锁。

STL模拟题

• 在模拟式需要重载set的()运算，重载的含义为：在 abs(a - b) <= k 存在的情况下不插入b，否则按插入b后升序排序。
• set的其他用法百度随处可搜，本题需要掌握erase()和find()。

#include<bits/stdc++.h>
#define INF 0x3f3f3f3f
#define DOF 0x7f7f7f7f
#define endl '\n'
#define mem(a,b) memset(a,b,sizeof(a))
#define debug(case,x); cout<<case<<"  : "<<x<<endl;
#define IO ios::sync_with_stdio(false);cin.tie(0);cout.tie(0)
typedef long long ll;
using namespace std;
const int maxn = 5e5 + 10;
int m, k;
struct cmp {
    bool operator ()(const int& a, const int& b)const {
        if(abs(a - b) <= k)
            return false;
        return a < b;
    }
};

set<int, cmp>st;
int main() {
    IO;
    string str;
    cin >> m >> k;
    while(m--) {
        int t;
        cin >> str >> t;
        if(str == "add")
            st.insert(t);
        else if(str == "del")
            st.erase(t);
        else {
            if(st.find(t) != st.end())
                cout << "Yes\n";
            else
                cout << "No\n";
        }

    }
}