Advertisement
smj007

Untitled

Mar 6th, 2024
588
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.95 KB | None | 0 0
  1. class SeatManager:
  2.  
  3.     def __init__(self, n: int):
  4.         self.minheap = [i for i in range(1, n+1)]
  5.         heapq.heapify(self.minheap)        
  6.  
  7.     def reserve(self) -> int:
  8.         top = heapq.heappop(self.minheap)
  9.         return top
  10.  
  11.     def unreserve(self, seatNumber: int) -> None:
  12.         heapq.heappush(self.minheap, seatNumber)        
  13.  
  14.  
  15. # Your SeatManager object will be instantiated and called as such:
  16. # obj = SeatManager(n)
  17. # param_1 = obj.reserve()
  18. # obj.unreserve(seatNumber)
  19.  
  20.  
  21. class SeatManager:
  22.  
  23.     def __init__(self, n: int):
  24.         self.marker = 1
  25.         self.minheap = []
  26.  
  27.     def reserve(self) -> int:
  28.         if self.minheap:
  29.             seat_number = heapq.heappop(self.minheap)
  30.             return seat_number
  31.  
  32.         seat_number = self.marker
  33.         self.marker += 1
  34.         return seat_number
  35.  
  36.     def unreserve(self, seatNumber: int) -> None:
  37.         heapq.heappush(self.minheap, seatNumber)
  38.        
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement