指纹锁(set)

题目描述

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

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

4 3
add 1
add 10
query 5
query 4

No
Yes

4 3
add 1
query 4
del 1
query 4

Yes
No

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)
#include<set>
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

int m , k;

struct cmp
{
	bool operator () (const int &a , const int &b)
	{
		if(abs(a - b) <= k)
			return false;
		return a < b;
	}
};

int main()
{
	set<int , cmp> s;
 	scanf("%d %d" , &m , &k);
 	while(m--)
 	{
		int num;
		char op[6];
		scanf("%s %d" , op , &num);
		
		if(op[0] == 'a')
			s.insert(num);
		else if(op[0] == 'd')
			s.erase(num);
		else
			printf("%s\n" , s.find(num) != s.end() ? "Yes" : "No");
	}
	return 0; 
}