对于100%的测试数据: 1 ≤ k,m ≤ 1000000 数据量较大,注意使用更快的输入输出方式。
这题也是模拟题,难点在于如何存数据,以及数据范围较大时,如果直接查找可能会超时。
还在犹豫要不要二分查找时,看到了下面的题解,我才知道set居然这么强大,确实在这里使用set非常的合适,set内置的find()函数,复杂度是**O(nlogn)**的,不用担心查找会超时。同时set是默认不同的才能插入,但这里不同的是,它定义的“不同”是在误差k内的。
所有需要像sort函数中的cmp一样,单独写一个结构体cmp用于set中的互异判断,真的太强大了,有了这个set,增删操作就是一个函数的问题,查询操作也是find一下就行了,太舒服了!
会用STL真的能节省很多时间。
参考题解:
using namespace std;
#include<bits/stdc++.h>
int m,k,x;
struct cmp {
bool operator () (const int& u, const int& v) const {
if (abs(u - v) <= k) return false;
return u < v;
}
};
set<int,cmp> s;
int main(){
ios::sync_with_stdio(false);
string op;
cin>>m>>k;
while(m--)
{
cin>>op>>x;
if(op == "add")
{
s.insert(x);
}
else if(op == "del")
{
s.erase(x);
}
else if(op == "query")
{
if(s.find(x) != s.end()) cout<<"Yes\n";
else cout<<"No\n";
}
}
return 0;
}