diff --git a/solutions/2336. Smallest Number in Infinite Set/2336.py b/solutions/2336. Smallest Number in Infinite Set/2336.py new file mode 100644 index 00000000000..5a85cc6fedc --- /dev/null +++ b/solutions/2336. Smallest Number in Infinite Set/2336.py @@ -0,0 +1,14 @@ +class SmallestInfiniteSet: + def __init__(self): + self.mini = 1 + self.heap = [] + + def popSmallest(self) -> int: + if self.heap: + return heapq.heappop(self.heap) + self.mini += 1 + return self.mini - 1 + + def addBack(self, num: int) -> None: + if num < self.mini and num not in self.heap: + heapq.heappush(self.heap, num) \ No newline at end of file