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

最新推荐文章于 2025-08-04 20:35:07 发布

ベ断桥烟雨ミ800

最新推荐文章于 2025-08-04 20:35:07 发布

阅读量265

点赞数 11

CC 4.0 BY-SA版权

文章标签： leetcode 算法 java TreeSet

本文链接：https://blog.youkuaiyun.com/qq_56388166/article/details/134697162

文章描述了一个编程问题，需要实现一个名为SmallestInfiniteSet的类，用于处理无限集中的正整数。类包含popSmallest方法移除并返回最小数，addBack方法在数不在集中时添加。使用TreeSet数据结构来高效管理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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]

解释
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 次

题解


class SmallestInfiniteSet {
    // TreeSet
    public TreeSet<Integer> set;
    // 初始化 SmallestInfiniteSet 对象以包含 所有 正整数。
    public SmallestInfiniteSet() {
        // 初始化
        set = new TreeSet<>();
        // z正整数范围1~1000
        for(int i = 1;i<=1000;i++){
            set.add(i);
        }
    }
    //  移除 并返回该无限集中的最小整数。
    public int popSmallest() {
        // 获取最小整数
        int ans = set.pollFirst();
        // 返回最小整数
        return ans; 
    }
    // 如果正整数 num 不 存在于无限集中，则将一个 num 添加 到该无限集最后。
    public void addBack(int num) {
        set.add(num);
    }
}

/**
 * Your SmallestInfiniteSet object will be instantiated and called as such:
 * SmallestInfiniteSet obj = new SmallestInfiniteSet();
 * int param_1 = obj.popSmallest();
 * obj.addBack(num);
 */