Leetcode 2336. Smallest Number in Infinite Set [Python]

最新推荐文章于 2024-05-29 11:19:10 发布

Case_CaiNiao

最新推荐文章于 2024-05-29 11:19:10 发布

阅读量353

点赞数

分类专栏： Leetcode学习记录文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/sinat_30403031/article/details/126562511

版权

Leetcode学习记录专栏收录该内容

268 篇文章 1 订阅

订阅专栏

系统设计题，简单的用个list。用in来判断第二个addback函数的条件是否成立。bisect的insort_left函数保证add back数字后，set里的元素依然是顺序的。trace smallest则只需要拿出list的头元素就可以。

class SmallestInfiniteSet:

    def __init__(self):
        self.Set = [i for i in range(1, 1001)]

    def popSmallest(self) -> int:
        if self.Set:
            v = self.Set[0]
            self.Set.pop(0)
        return v
        
    def addBack(self, num: int) -> None:
        if num in self.Set:
            return
        bisect.insort_left(self.Set, num)