[LeetCode 75] Medium - 2336. Smallest Number in Infinite Set

2336. Smallest Number in Infinite Set

 

class SmallestInfiniteSet:

    def __init__(self):
        self.res = [True] * 1001
        

    def popSmallest(self) -> int:
        for i in range(1, len(self.res)):
            if self.res[i] == True:
                self.res[i] = False
                return i
        

    def addBack(self, num: int) -> None:
        self.res[num] = True
        


# Your SmallestInfiniteSet object will be instantiated and called as such:
# obj = SmallestInfiniteSet()
# param_1 = obj.popSmallest()
# obj.addBack(num)