From 71d77923c80cb3c51ad3a52fb4eea7c2aa0e58c3 Mon Sep 17 00:00:00 2001 From: TheRootDaemon Date: Wed, 17 Sep 2025 21:30:31 +0530 Subject: [PATCH] feat: Python solution for LeetCode 2336 --- .../2336. Smallest Number in Infinite Set/2336.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 solutions/2336. Smallest Number in Infinite Set/2336.py 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