牛客 指纹锁 （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”，表示该人是否可以打开指纹锁。

输入4 3add 1add 10query 5query 4

输出NoYes

输入4 3add 1query 4del 1query 4

输出YesNo

输入6 3add 10query 10add 5query 5del 7 //系统将指纹10和指纹5全部删除query 8

输出

YesYesNo

备注:对于100%的测试数据：1 ≤ k,m ≤ 1000000数据量较大，注意使用更快的输入输出方式。

题目分析

• 毫无疑问这个题目用set是最佳选择，因为set容器不允许出现相同的，而且其内部排序是一颗红黑树，速度快且便捷。
• 在这里我们首先需要重载（）运算符，自定义其排序规则，并且用set去对一个区间进行合并，绝对值在k之间的数是不会被插入，也不会被删除，这样就维护了唯一性，如果大于k那么就按照升序排列
• 所以每次我们只需要判断插入的点和删除的点是不是在某一个点的合并范围，两个点之间的距离为k的时候直接返回false，表示不用装进去了，就不要这个数了。

AC代码

#include <bits/stdc++.h>
using namespace std;
int m, k, x;
struct cmp
{ //自定义比较重载下set的圆括号比较，把绝对值在k之间的数定义false，那么erase就可以直接删掉。insert也不会插入。
    bool operator()(const int u, const int v) const
    {
        if (abs(u - v) <= k)
            return false;
        return u < v;
    }
};
set<int, cmp> se;
int main()
{
    scanf("%d%d", &m, &k);
    while (m--)
    {
        string s;
        cin >> s;
        scanf("%d", &x);
        if ("add" == s)
            se.insert(x);
        else if ("del" == s)
            se.erase(x);
        else if (se.find(x) != se.end())//如果到末尾都没找到，说明符合题目要求
            puts("Yes");
        else
            puts("No");
    }
    return 0;
}