题目
链接:https://ac.nowcoder.com/acm/contest/28886/1014
来源:牛客网时间限制:C/C++ 1秒,其他语言2秒
空间限制:C/C++ 262144K,其他语言524288K
64bit IO Format: %lld题目描述
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莫属;
主角上场:lower_bound和upper_bound为二分法查找元素,其时间复杂度为O(log n)。
lower_bound返回大于等于该元素的第一个元素
upper_bound返回大于该元素的第一个元素
如果找不到,返回末尾的迭代器
以下代码可以用来测试upper_bound和lower_bound的区别
#include <iostream>
#include <algorithm>
using namespace std;
int a[] = {
0,0,0,0 };
void p(int* x)
{
if (x == a + 4)
cout << "NULL" << endl;
else
{
cout << *x << ' ' << x - a << endl;
}
}
int main()
{
p(upper_bound(a, a + 4, 1));
p(lower_bound(a, a + 4, 1));
return 0;
}
可以再以非递减排列的:
数组lower_bound(数组名, 数组名 + size, val)
vector
vector<int> vec;
lower_bound(vec.begin(), vec.end(), val);
以及:
map
set<int> s;
s.lower_bound(2);
set
map<int, int> m;
m.upper_bound(2);
代码
告诫:一定不要使用cin和cout
#include <algorithm>
#include <iostream>
#include <set>
#include <stdio.h>
using namespace std;
int m, k;
set<int>st;
int main()
{
//ios::sync_with_stdio(false);
//cin.tie(0);
//cout.tie(0);
cin >> m >> k;
while (m--)
{
char s[12];
int data;
scanf("%s%d",s,&data);
auto it = st.lower_bound(data - k);
auto tmp = st.lower_bound(data - k);
switch (s[0])
{
case 'a'://插入
if (it != st.end() && *it <= data + k)
break;
st.insert(data);
break;
case 'd'://删除
while (it != st.end() && *it <= data + k)
{
tmp = it;//值得注意
it++;
st.erase(tmp);
if (st.empty() || it == st.end())//值得注意
break;
}
break;
case 'q'://判断
if (it != st.end() && *it <= data + k)
printf("Yes\n");
else
printf("No\n");
}
}
return 0;
}