"Seat Reservation Manager" solution

Author: lancelote

src/seat_reservation_manager.py

@@ -0,0 +1,19 @@
+import heapq
+
+
+class SeatManager:
+    def __init__(self, n: int):
+        self.i = 1
+        self._unreserved: list[int] = []
+
+    def reserve(self) -> int:
+        if self._unreserved and self._unreserved[0] < self.i:
+            result = heapq.heappop(self._unreserved)
+        else:
+            result = self.i
+            self.i += 1
+
+        return result
+
+    def unreserve(self, seat_number: int) -> None:
+        heapq.heappush(self._unreserved, seat_number)

tests/test_seat_reservation_manager.py

@@ -0,0 +1,17 @@
+from src.seat_reservation_manager import SeatManager
+
+
+def test_seat_manager():
+    sm = SeatManager(5)
+
+    assert sm.reserve() == 1
+    assert sm.reserve() == 2
+
+    sm.unreserve(2)
+
+    assert sm.reserve() == 2
+    assert sm.reserve() == 3
+    assert sm.reserve() == 4
+    assert sm.reserve() == 5
+
+    sm.unreserve(5)