题目描述
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 3
add 1
add 10
query 5
query 4
输出:
No
Yes
思路
可以考虑用set,题目中提到的指纹不重复比较符合set的性质
(参考:题解)
#include <bits/stdc++.h>
using namespace std;
int m,k,x;
char s[10];
struct cmp{
bool operator () (const int & x,const int & y) const {
if(abs(x-y)<=k) return false;
return x < y;
}
};
set<int,cmp> se;
int main(){
// scanf("%d %d",&m,&k);
cin>>m;
cin>>k;
while(m--){
//scanf("%s",s);scanf("%d",&x);
cin>>s;cin>>x;
if(s[0]=='a') se.insert(x);
else if(s[0]=='d') se.erase(x);
else{
if (se.find(x)!=se.end()) puts("Yes");
else puts("No");
}
}
return 0;
}
需要注意的是,这个题用cin,cout会超时,原因:
scanf较cout快
补充:set重载
set重载参考
set中判断不等的办法是!a<b&&!a>b