指纹锁（set 自定义比较用法）

LINK、

题目

样例

示例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
数据量较大，注意使用更快的输入输出方式。

代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int N=1e5+10;
int n,k;
struct cmp{//自定义比较
	bool operator()(int x,int y){
		if(abs(x-y)<=k)return false;
		return x<y;
	}
};
set<int,cmp>s;
int main() {
//     ios::sync_with_stdio(0);cin.tie(0);cout.tie(0);
	scanf("%d%d",&n,&k);
	int x;char ss[10];
	while(n--){
		scanf("%s %d",ss,&x);
		if(ss[0]=='a'){//add
			if(s.find(x)==s.end())s.insert(x);
		}else if(ss[0]=='d'){//del
			s.erase(x);
		}else{//query
			if(s.find(x)!=s.end())printf("Yes\n");
			else printf("No\n");
		}
	}
	return 0;
}