指纹锁---带你见识set的强大

最新推荐文章于 2022-05-25 14:14:43 发布

原创最新推荐文章于 2022-05-25 14:14:43 发布 · 1.6k 阅读

0 ·

CC 4.0 BY-SA版权

牛客网专栏收录该内容

21 篇文章

订阅专栏

本文介绍了一种使用C++ STL中的Set数据结构解决大数据量查找问题的方法，通过自定义比较器实现误差范围内的元素查找，有效避免了超时问题，大幅提升了查找效率。

在这里插入图片描述

对于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;
}