指纹锁 来源:牛客网

题目

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