其他系列文章导航
文章目录
前言
这是力扣的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] 解释 SmallestInfiniteSet smallestInfiniteSet = new SmallestInfiniteSet(); smallestInfiniteSet.addBack(2); // 2 已经在集合中，所以不做任何变更。 smallestInfiniteSet.popSmallest(); // 返回 1 ，因为 1 是最小的整数，并将其从集合中移除。 smallestInfiniteSet.popSmallest(); // 返回 2 ，并将其从集合中移除。 smallestInfiniteSet.popSmallest(); // 返回 3 ，并将其从集合中移除。 smallestInfiniteSet.addBack(1); // 将 1 添加到该集合中。 smallestInfiniteSet.popSmallest(); // 返回 1 ，因为 1 在上一步中被添加到集合中， // 且 1 是最小的整数，并将其从集合中移除。 smallestInfiniteSet.popSmallest(); // 返回 4 ，并将其从集合中移除。 smallestInfiniteSet.popSmallest(); // 返回 5 ，并将其从集合中移除。
提示：
1 <= num <= 1000- 最多调用
popSmallest和addBack方法 共计1000次
二、题解
这题的关键点是始终要保证无限集合是连续的 。无限集合的范围可以认为是从 1 到正无穷大，并且都是正整数。
这道我是用TreeSet 和一个min 变量来维护这个无限集合。为什么用TreeSet ，因为TreeSet支持维护元素的自然顺序。
TreeSet：小于min的有序集合。
min：有序集合的最小值。
添加元素的时候分为两种情况：
-
添加元素的时候如果添加的值大于等于无限集合中的最小值
min，就不要添加，因为无限集合是连续的，添加的元素在无限集合中已经存在。（简单点说：比min还大的数不用加，说明已经存在了） -
添加的元素如果小于无限集合的最小值 min 也不能直接添加，如果贸然添加会导致无限集合不连续，只需要把它添加到有序集合 TreeSet 中即可， TreeSet 中存放的值都是小于 min 的。所以要加在TreeSet中，要靠TreeSet来维护。
删除元素的时候：
- 删除的时候先判断有序集合 TreeSet 是否为空，如果不为空，说明存在比 min 还小的元素，直接从 TreeSet 中删除。否则就从有序集合中删除 min ，删除之后 min 值要加 1 。
三、代码
java
class SmallestInfiniteSet {
TreeSet<Integer> set = new TreeSet<>();
int min = 1;
public SmallestInfiniteSet() {}
public int popSmallest() {
if (set.isEmpty()) {
return min++;//先返回，再++
}
return set.pollFirst();//弹出set的第一个元素，并删除
}
public void addBack(int num) {
if (num < min) {//大于的话，说明存在了
set.add(num);
}
}
}四、总结
使用TreeSet和min变量来维护一个无限集合，保证集合的连续性。添加元素时，若元素大于等于min，则不添加；若元素小于min，则将其添加到TreeSet中。删除元素时，先判断TreeSet是否为空，若不为空，则从TreeSet中删除元素；若为空，则将min值加1。该算法能够高效地添加和删除元素，并保持集合的连续性。
该算法还可以用优先队列（小根堆）+ hash表解题，比较优秀。