NC 17508 STL

最新推荐文章于 2021-07-16 11:10:23 发布

SHOHOKUKU

最新推荐文章于 2021-07-16 11:10:23 发布

阅读量91

点赞数

分类专栏：数据结构

本文链接：https://blog.csdn.net/neweryyy/article/details/106933255

版权

数据结构专栏收录该内容

203 篇文章 1 订阅

订阅专栏

题意

传送门 NC 17508

题解

版本一

若存在数字 $n\in [x-k,x+k]$ ，则 $lower\_bound(x-k)\ !=\ upper\_bound(x+k)$ ，用 $s e t$ 维护指纹锁即可。

#include <bits/stdc++.h>
using namespace std;
int main()
{
    set<int> key;
    int m, k;
    scanf("%d%d", &m, &k);
    for (int i = 0; i < m; i++)
    {
        char op[10];
        int x;
        scanf(" %s %d", op, &x);
        if (op[0] == 'a')
        {
            if (key.lower_bound(x - k) == key.upper_bound(x + k))
            {
                key.insert(x);
            }
        }
        else if (op[0] == 'd')
        {
            key.erase(key.lower_bound(x - k), key.upper_bound(x + k));
        }
        else
        {
            puts(key.lower_bound(x - k) == key.upper_bound(x + k) ? "No" : "Yes");
        }
    }
    return 0;
}

版本二

重载 $s e t$ 的比较运算，使区间 $[x - k, x + k]$ 内的点看做相同的点。

#include <bits/stdc++.h>
using namespace std;
int m, k;
struct cmp
{
    bool operator()(const int &a, const int &b) const
    {
        if (abs(a - b) <= k) return false;
        return a < b;
    }
};

int main()
{
    set<int, cmp> key;
    scanf("%d%d", &m, &k);
    for (int i = 0; i < m; i++)
    {
        char op[10];
        int x;
        scanf(" %s %d", op, &x);
        if (op[0] == 'a')
        {
            key.insert(x);
        }
        else if (op[0] == 'd')
        {
            key.erase(x);
        }
        else
        {
            puts(key.find(x) == key.end() ? "No" : "Yes");
        }
    }
    return 0;
}