feat: 1845_seat_reservation_manager

This commit is contained in:
SquidSpirit 2025-09-20 02:29:27 +08:00
parent 677ac9b9ff
commit 163aeafd47
3 changed files with 40 additions and 0 deletions

View File

@ -0,0 +1,7 @@
# This file is automatically @generated by Cargo.
# It is not intended for manual editing.
version = 4
[[package]]
name = "seat_reservation_manager"
version = "0.1.0"

View File

@ -0,0 +1,6 @@
[package]
name = "seat_reservation_manager"
version = "0.1.0"
edition = "2024"
[dependencies]

View File

@ -0,0 +1,27 @@
use std::{cmp::Reverse, collections::BinaryHeap};
pub struct SeatManager {
unreserved_seats: BinaryHeap<Reverse<i32>>,
}
impl SeatManager {
pub fn new(n: i32) -> Self {
Self {
unreserved_seats: BinaryHeap::from_iter((0..n).map(|i| Reverse(i))),
}
}
pub fn reserve(&mut self) -> i32 {
if let Some(Reverse(smallest_available_seat)) = self.unreserved_seats.peek() {
let result = *smallest_available_seat + 1;
self.unreserved_seats.pop();
result
} else {
-1
}
}
pub fn unreserve(&mut self, seat: i32) {
self.unreserved_seats.push(Reverse(seat - 1));
}
}