1845. 座位预约管理系统
题目描述:
请你设计一个管理 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
次。
解题分析及思路:
方法:队列
思路:
为 SeatManager
创建一个数组,用来记录座位是否被预约。并且创建一个变量 first
,用来记录当前可预约座位的最小下标。
在 reserve
中,需要做两件事:
- 返回原始的最小下标,并且将下标对应的值置为
true
,表示该座位被预约。 - 更新
first
,使其指向下一个可以预约的座位。
在 unreserve
中,只需要将座位对应的值置为 false
,并且更新 first
,使其指向最小可以预约的座位下标。
注意:座位编号 = 下标值 + 1
type SeatManager struct {
stack []bool
first int
}
func Constructor(n int) SeatManager {
return SeatManager{
make([]bool, n),
0,
}
}
func (this *SeatManager) Reserve() int {
this.stack[this.first] = true
var num = this.first + 1
for i := this.first; i < len(this.stack); i++ {
this.first = i
if !this.stack[i] {
break
}
}
return num
}
func (this *SeatManager) Unreserve(seatNumber int) {
index := seatNumber - 1
this.stack[index] = false
if index < this.first {
this.first = index
}
}
复杂度:
- 时间复杂度:O(n)
- 空间复杂度:O(1)
执行结果:
- 执行耗时:853 ms,击败了8.70% 的Go用户
- 内存消耗:30.4 MB,击败了30.44% 的Go用户