问题描述:
在考场里,一排有 N
个座位,分别编号为 0, 1, 2, ..., N-1
。
当学生进入考场后,他必须坐在能够使他与离他最近的人之间的距离达到最大化的座位上。如果有多个这样的座位,他会坐在编号最小的座位上。(另外,如果考场里没有人,那么学生就坐在 0 号座位上。)
返回 ExamRoom(int N)
类,它有两个公开的函数:其中,函数 ExamRoom.seat()
会返回一个 int
(整型数据),代表学生坐的位置;函数 ExamRoom.leave(int p)
代表坐在座位 p
上的学生现在离开了考场。每次调用 ExamRoom.leave(p)
时都保证有学生坐在座位 p
上。
示例:
输入:["ExamRoom","seat","seat","seat","seat","leave","seat"], [[10],[],[],[],[],[4],[]] 输出:[null,0,9,4,2,null,5] 解释: ExamRoom(10) -> null seat() -> 0,没有人在考场里,那么学生坐在 0 号座位上。 seat() -> 9,学生最后坐在 9 号座位上。 seat() -> 4,学生最后坐在 4 号座位上。 seat() -> 2,学生最后坐在 2 号座位上。 leave(4) -> null seat() -> 5,学生最后坐在 5 号座位上。
提示:
1 <= N <= 10^9
- 在所有的测试样例中
ExamRoom.seat()
和ExamRoom.leave()
最多被调用10^4
次。 - 保证在调用
ExamRoom.leave(p)
时有学生正坐在座位p
上。
解题思路:
- 如何构建离其他人的距离最大: 就是找两个点的中点位置。
- 特殊情况:没有人,那么我们从0开始。
- 实现: 这里需要有序性,所以使用set来实现。
代码实现(C++):
class ExamRoom {
int n;
set<int> s;
public:
ExamRoom(int n) {
this->n = n;
}
int seat() {
if (s.empty()) {
s.insert(0);
return 0;
}
// 应该坐的位置
int res = 0;
// 上一个位置
int pre = -1;
// 最大距离
int maxDist = 0;
for (cur : s) {
// 当前距离为:两个座位之间距离的一半
int dist = (cur - pre) >> 1;
if (dist > maxDist) {
maxDist = dist;
res = pre == -1 ? 0 : pre + dist;
}
pre = cur;
}
// 最后一个位置的特殊情况
if (n - 1 - pre > maxDist) {
res = n - 1;
}
s.insert(res);
return res;
}
void leave(int p) {
// 离开座位即删除
s.erase(p);
}
};