链接: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;
}