题解 | #指纹锁#

链接：https://ac.nowcoder.com/acm/problem/17508
来源：牛客网

题目描述

思路：set函数+重载运算符

注：set函数内元素如果不是结构体，可以重载()运算符；如果是结构体，可以讲比较函数写在结构体之内，重载<运算符

本题使用重载()运算符，实现判断是否需要插入，一次性若干个删除，以及查询范围内是否存在相关元素的三个操作

add x对应set里的insert()函数

delete x对应set里的erase()函数

query x对应set里的find()函数，当查找不到时，其值=end()

本题主要是模拟一个过程，不需要考虑其他算法

代码如下

#include<stdio.h>
#include<bits/stdc++.h>
#pragma GCC optimize(2) //开启O2优化，虽然似乎只提升了10~20ms
using namespace std;
char s[10];
int m,k,num;
struct comp{
    bool operator () (const int &a,const int &b)const{ //不是结构体，重载()运算符
        if(abs(a-b)<=k) return false; //return false表明现在输入的数与之前输入的数一样，不加入set里 
        else return a<b; //表明输入的数与之前的数不相同，就按照从小到大进行排序 
    }
}; //这个重载运算符在insert()和erase()中被使用，用来判断是否需要插入，一次性若干个删除，查询范围内是否存在相关元素
set<int,comp>st; //创建一个set函数 
inline int read() //快读函数，提升了200ms左右
{
    int x=0,f=1;
    char ch=getchar();
    while(ch<'0'||ch>'9')
    {
        if(ch=='-')
            f=-1;
        ch=getchar();
    }
    while(ch>='0' && ch<='9')
        x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int main() {
    m=read();k=read();
    //scanf("%d %d",&m,&k);
    while(m--) {
        scanf("%s",s);
        num=read();
        //scanf("%d",&num);
        if(s[0]=='a') {
            st.insert(num); //执行插入操作
        }
        if(s[0]=='d') {
            st.erase(num); //执行删除操作
        }
        if(s[0]=='q') {
            if(st.find(num)==st.end()) printf("No\n"); //执行查找操作，set里的find()函数查找不到此元素时，返回值为end() 
            else printf("Yes\n");
        }
    }
    return 0;
}