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);
}
};