题目:
请你设计一个管理 n
个座位预约的系统,座位编号从 1
到 n
。
请你实现 SeatManager
类:
SeatManager(int n)
初始化一个SeatManager
对象,它管理从1
到n
编号的n
个座位。所有座位初始都是可预约的。int reserve()
返回可以预约座位的 最小编号 ,此座位变为不可预约。void unreserve(int seatNumber)
将给定编号seatNumber
对应的座位变成可以预约。
示例 1:
输入: ["SeatManager", "reserve", "reserve", "unreserve", "reserve", "reserve", "reserve", "reserve", "unreserve"] [[5], [], [], [2], [], [], [], [], [5]] 输出: [null, 1, 2, null, 2, 3, 4, 5, null] 解释: SeatManager seatManager = new SeatManager(5); // 初始化 SeatManager ,有 5 个座位。 seatManager.reserve(); // 所有座位都可以预约,所以返回最小编号的座位,也就是 1 。 seatManager.reserve(); // 可以预约的座位为 [2,3,4,5] ,返回最小编号的座位,也就是 2 。 seatManager.unreserve(2); // 将座位 2 变为可以预约,现在可预约的座位为 [2,3,4,5] 。 seatManager.reserve(); // 可以预约的座位为 [2,3,4,5] ,返回最小编号的座位,也就是 2 。 seatManager.reserve(); // 可以预约的座位为 [3,4,5] ,返回最小编号的座位,也就是 3 。 seatManager.reserve(); // 可以预约的座位为 [4,5] ,返回最小编号的座位,也就是 4 。 seatManager.reserve(); // 唯一可以预约的是座位 5 ,所以返回 5 。 seatManager.unreserve(5); // 将座位 5 变为可以预约,现在可预约的座位为 [5] 。
提示:
1 <= n <= 105
1 <= seatNumber <= n
- 每一次对
reserve
的调用,题目保证至少存在一个可以预约的座位。 - 每一次对
unreserve
的调用,题目保证seatNumber
在调用函数前都是被预约状态。 - 对
reserve
和unreserve
的调用 总共 不超过105
次。
解题:
为了管理座位预约信息,我们需要一种方式来快速识别最小的未预约座位编号,并且能够在座位被释放时,迅速将其标记为可预约。解决这个问题的策略是使用一个优先队列(PriorityQueue
)来管理未预约的座位编号。PriorityQueue
能够保证在任何时候都能以对数时间复杂度取出最小元素。以下是具体的解题步骤:
- 使用
PriorityQueue
来存储所有可预约的座位编号。 - 当执行
reserve
动作时,从PriorityQueue
中取出并移除具有最小编号的座位。 - 当执行
unreserve
动作时,将释放的座位编号加回PriorityQueue
中。
代码:
import java.util.PriorityQueue;
public class SeatManager {
private PriorityQueue<Integer> availableSeats;
public SeatManager(int n) {
// 初始化可预约座位的优先队列
availableSeats = new PriorityQueue<>();
// 将所有座位编号添加到优先队列中
for (int i = 1; i <= n; i++) {
availableSeats.offer(i);
}
}
public int reserve() {
// 从优先队列中拿出最小可预约座位编号
return availableSeats.poll();
}
public void unreserve(int seatNumber) {
// 将取消预约的座位编号放回到优先队列中
availableSeats.offer(seatNumber);
}
}
知识点解析:
-
优先队列
PriorityQueue
: Java 中的PriorityQueue
是一个基于优先级堆的数据结构,它允许快速访问队列中的最小元素。默认情况下,元素按自然排序顺序存储,即数字较小的元素拥有更高的优先级。 -
构造函数: Java 中的构造函数用于创建对象并为对象成员变量初始化值。
-
泛型:
PriorityQueue
使用泛型来指定队列中允许的元素类型,例如PriorityQueue<Integer>
表明队列中的元素类型为Integer
。 -
自动装箱和拆箱: Java 允许自动在基本类型和包装器类型(如
int
和Integer
)之间转换。 -
方法: Java 类中的方法(如
reserve
和unreserve
)执行特定的操作。reserve
返回并移除队列的最小元素,unreserve
将一个元素重新加入队列。
知识点 | 说明 | 代码示例 |
---|---|---|
优先队列 PriorityQueue | 以对数时间复杂度访问最小元素的数据结构 | new PriorityQueue<Integer>(); |
构造函数 | 创建对象并初始化成员变量 | public SeatManager(int n) { ... } |
泛型 | 在集合中指定元素类型 | PriorityQueue<Integer> |
自动装箱和拆箱 | 基本类型和包装器类型间自动转换 | availableSeats.offer(seatNumber); |
方法 | 执行类中定义的特定操作 | public int reserve() { ... } |
2024.5.18