NC17508 指纹锁 set 一次erase删除多个元素 重载比较

链接：https://ac.nowcoder.com/acm/problem/17508
来源：牛客网

题目描述
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$一次$erase$删除多个元素!!!
惊呆了!!!

struct cmp {
	//神奇的比较函数,可以然erase一次删除多个元素
	bool operator () (const int x, const int y) {
		if(abs(x-y) <= K) return 0;
		else return x < y;
	}
} ;

set<int, cmp> seta;

完整代码

#define debug
#ifdef debug
#include <time.h>
#include "/home/majiao/mb.h"
#endif

#include <iostream>
#include <algorithm>
#include <vector>
#include <string.h>
#include <map>
#include <set>
#include <stack>
#include <stdlib.h>
#include <queue>
#include <math.h>

#define MAXN ((int)1e5+7)
#define ll long long int
#define INF (0x7f7f7f7f)
#define fori(lef, rig) for(int i=lef; i<=rig; i++)
#define forj(lef, rig) for(int j=lef; j<=rig; j++)
#define fork(lef, rig) for(int k=lef; k<=rig; k++)
#define QAQ (0)

using namespace std;

#ifdef debug
#define show(x...) \
	do { \
		cout << "\033[31;1m " << #x << " -> "; \
		err(x); \
	} while (0)

void err() { cout << "\033[39;0m" << endl; }
template<typename T, typename... A>
void err(T a, A... x) { cout << a << ' '; err(x...); }
#endif

#ifndef debug
namespace FIO {
	template <typename T>
	inline void read(T& x) {
		int f = 1; x = 0;
		char ch = getchar();

		while (ch < '0' || ch > '9') 
			{ if (ch == '-') f = -1; ch = getchar(); }
		while (ch >= '0' && ch <= '9') 
			{ x = x * 10 + ch - '0'; ch = getchar(); }
		x *= f;
	}
};
using namespace FIO;
#endif

int n, m, Q, K;
char cmd[8];

struct cmp {
	//神奇的比较函数,可以然erase一次删除多个元素
	bool operator () (const int x, const int y) {
		if(abs(x-y) <= K) return 0;
		else return x < y;
	}
} ;

set<int, cmp> seta;

int main() {
#ifdef debug
	freopen("test", "r", stdin);
	clock_t stime = clock();
#endif
#if 1
	scanf("%d %d ", &Q, &K);
	while(Q--) {
		scanf("%s %d ", cmd, &m);
		if(cmd[0] == 'a') {
			seta.insert(m);
		} else if(cmd[0] == 'q') {
			if(seta.find(m) != seta.end())
				printf("Yes\n");
			else
				printf("No\n");
		} else {
			seta.erase(m); //一次erase就能删除多个元素,神奇!!!
		}
	}
#else
	K = 3;
	seta.insert(5);
	seta.insert(10);
	cout << "find 7 : " << *seta.find(7) << endl;
	cout << "before del : ";
	for(auto it : seta) cout << "[" << it << "]";
		cout << endl;
	seta.erase(7); //太神奇了, 一次erase可以删除5和10两个元素!!!
	cout << "after del : ";
	for(auto it : seta) cout << "[" << it << "]";
		cout << endl;

#endif



#ifdef debug
	clock_t etime = clock();
	printf("rum time: %lf 秒\n",(double) (etime-stime)/CLOCKS_PER_SEC);
#endif 
	return 0;
}