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

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

输出
No
Yes

示例2

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

输出
Yes
No

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

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

//这个题主要是写对set的自定义的方法，要注意的是，运算符的重载不能像sort()那种直接写，而是要在结构体内重载。

#include <iostream>
#include <string>
#include <set>
using namespace std;
#define IOS ios::sync_with_stdio(false); cin.tie(0); cout.tie(0);
//代码预处理区

const int MAX = 1e6 + 7;
int m, k;
struct cmp {
    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() {
    IOS;
    cin >> m >> k;
    while (m--) {
        string s; cin >> s;
        int x; cin >> x;
        if ("add" == s) se.insert(x);
        else if ("del" == s) se.erase(x);
        else
            if (se.find(x) != se.end()) cout << "Yes" << endl;
            else cout << "No" << endl;
    }
    return 0;
}

标签：set,指纹锁,int,指纹,运算符,add,query,Yes
来源： https://blog.csdn.net/weixin_45465598/article/details/110728323