1.LeetCode:2336. 无限集中的最小数字

原题链接

        现有一个包含所有正整数的集合 [1, 2, 3, 4, 5, …] 。

        实现 SmallestInfiniteSet 类：

        SmallestInfiniteSet() 初始化 SmallestInfiniteSet 对象以包含 所有 正整数。

        int popSmallest() 移除 并返回该无限集中的最小整数。

        void addBack(int num) 如果正整数 num 不 存在于无限集中，则将一个 num 添加 到该无限集中。

        示例：

        输入

        [“SmallestInfiniteSet”, “addBack”, “popSmallest”, “popSmallest”, “popSmallest”, “addBack”, “popSmallest”, “popSmallest”, “popSmallest”]

        [[], [2], [], [], [], [1], [], [], []]

        输出

        [null, null, 1, 2, 3, null, 1, 4, 5]

        提示：

        1 <= num <= 1000

        最多调用 popSmallest 和 addBack 方法 共计 1000 次

        

        由于最多调用pop和add操作共1000次，所以最多也就删除集合中前1000个元素，直接用一个set来插入1~1000，然后按照函数来操作即可

class SmallestInfiniteSet {
    set<int> s;
public:
    SmallestInfiniteSet() {
        s.clear();
        for(int i=1;i<=1000;++i){
            s.insert(i);
        }
    }
    
    int popSmallest() {
        int ans=*s.begin();
        s.erase(s.begin());
        return ans;
    }   
    
    void addBack(int num) {
        s.insert(num);
    }
};
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20