Leetcode 2336. Smallest Number in Infinite Set [Python]

系统设计题，简单的用个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)
    
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16