feat: 1845_seat_reservation_manager
This commit is contained in:
parent
677ac9b9ff
commit
163aeafd47
7
1845_seat_reservation_manager/Cargo.lock
generated
Normal file
7
1845_seat_reservation_manager/Cargo.lock
generated
Normal 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"
|
6
1845_seat_reservation_manager/Cargo.toml
Normal file
6
1845_seat_reservation_manager/Cargo.toml
Normal file
@ -0,0 +1,6 @@
|
||||
[package]
|
||||
name = "seat_reservation_manager"
|
||||
version = "0.1.0"
|
||||
edition = "2024"
|
||||
|
||||
[dependencies]
|
27
1845_seat_reservation_manager/src/lib.rs
Normal file
27
1845_seat_reservation_manager/src/lib.rs
Normal 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));
|
||||
}
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user