Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- class SeatManager:
- def __init__(self, n: int):
- self.minheap = [i for i in range(1, n+1)]
- heapq.heapify(self.minheap)
- def reserve(self) -> int:
- top = heapq.heappop(self.minheap)
- return top
- def unreserve(self, seatNumber: int) -> None:
- heapq.heappush(self.minheap, seatNumber)
- # Your SeatManager object will be instantiated and called as such:
- # obj = SeatManager(n)
- # param_1 = obj.reserve()
- # obj.unreserve(seatNumber)
- class SeatManager:
- def __init__(self, n: int):
- self.marker = 1
- self.minheap = []
- def reserve(self) -> int:
- if self.minheap:
- seat_number = heapq.heappop(self.minheap)
- return seat_number
- seat_number = self.marker
- self.marker += 1
- return seat_number
- def unreserve(self, seatNumber: int) -> None:
- heapq.heappush(self.minheap, seatNumber)
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement