Update Seat Reservation Manager.java

varunu28 · web-flow · commit bcc80bf2250a · 2023-11-08T05:46:39.000+05:30
diff --git a/Medium/Seat Reservation Manager.java b/Medium/Seat Reservation Manager.java
@@ -1,23 +1,21 @@
 class SeatManager {
 
-  PriorityQueue<Integer> pq;
-  int count;
+    private PriorityQueue<Integer> seats;
 
-  public SeatManager(int n) {
-    pq = new PriorityQueue<>();
-    count = 1;
-  }
-
-  public int reserve() {
-    if (pq.size() == 0) {
-      return count++;
+    public SeatManager(int n) {
+        this.seats = new PriorityQueue<>();
+        for (int i = 1; i <= n; i++) {
+            this.seats.add(i);
+        }
+    }
+    
+    public int reserve() {
+        return this.seats.poll();
+    }
+    
+    public void unreserve(int seatNumber) {
+        this.seats.add(seatNumber);
     }
-    return pq.remove();
-  }
-
-  public void unreserve(int seatNumber) {
-    pq.add(seatNumber);
-  }
 }
 
 /**
