第十五届中北大学算法与程序设计竞赛 F题集合操作

这题说实话比赛的时候都没开,赛后一看还不是太难,但是技巧性很强,而且我当真没思路,还是做题少,赛后看了一个大佬的代码,太巧啦真的佩服,自己整理了一下写一下心得。
题面很简单:
链接:https://ac.nowcoder.com/acm/contest/5188/F
来源:牛客网

有一个集合S,刚开始是空集,现在有3种操作。
1.往集合中添加x,如果集合中已经存在x,则不添加。
2.从集合中删除x,如果集合中不存在x,则不删除。
3.查询大于等于x且不在集合S中的最小的正整数、
这题一看就是卡时间的问题,那么问题来啦,怎么在O(logn)时间内查找到最小的那个整数呢,首先我们考虑肯定是要用到set集合啦,这个没得说,那么这题既然想找到第一个比他大的且不再集合中的元素,首先set就是有序的,那么就满足二分查找的条件,然后我们在读入的时候将所有元素都插入进去,并且将x+1也插入进去,然后遍历操作数组,在反向操作一下,这里是关键,为什么要两次进行而且反向操作就可以了呢,一位正常来说如果我们第一遍直接利用二分查询,这显然是不可行的,很简单这么想啊,比如你同时把x和x+1插入进去,在查询的时候x+1就会作为返回的结果,这显然不是我们想看到的,所以对于那些插入进来的元素我们在进行反向操作的时候删除他们,删除他们的时候插入进来这相当于什么就是我们把现在的集合转变为一个比已知集合元素大的最小外部元素,这里需要理解一下,确实很巧,因为你删除了这个元素,它就相当于在外部集合里,二分查找在set集合内置有一个函数,这里我以前是没有用到过的,今天学到啦。
AC代码:

#include<bits/stdc++.h>
using namespace std;
const int N=1e6+11;
int op[N],x[N];
set<int> se;
int main(){
    int n;
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d%d",&op[i],&x[i]);
        se.insert(x[i]);
        se.insert(x[i]+1);
    }
    for(int i=1;i<=n;i++){
        if(op[i]==1){
            if(se.count(x[i])) se.erase(x[i]);
        }else if(op[i]==2) se.insert(x[i]);
        else{
            set<int>::iterator it=se.lower_bound(x[i]);
            int ans=*it;
            printf("%d\n",ans);
        }
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值