题目描述:
在考场里,一排有 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
上。
想法:刚开始看到这个题目时想用数组来实现,后来一看提示发现N最大可以取到,过于太大,使用数组的话可能会太消耗内存。后来决定用ArrayList来作为考场座位,因为ArrayList是可以扩容的,并且leave(p时,只需调用remove方法即可,感觉会方便一些。但"距离最大化"这个问题,却一直没有思路,直到考到了这位大神的博客,才有了些许的想法。
"距离最大化"思路:用maxLen来记录当前的最大距离,并且分别计算出相邻两位考生之间的距离,若计算的距离大于maxLen,则更新maxLen的值,同时把新进来的考生安排在maxLen这个"区间"即可。
注:对于代码中的lowerBound方法如有不明白的,可以点我哦!
上代码:
class ExamRoom {
//用list来记录有哪些座位已经有考生在坐
List<Integer> list;
int N;
public ExamRoom(int N) {
list = new ArrayList<Integer>();
this.N = N;
}
public int seat() {
//考场无人,考生做"0"号座位
if(list.size() == 0){
list.add(0);
return 0;
}
//result用来记录考场要坐的座位号,maxLen记录最大距离
int result = 0, maxLen = list.get(0);
//计算相邻两位考生之间的距离,且若距离大于当前的maxLen,应该更新maxLen
for(int i = 0; i < list.size() - 1; i++){
int a = list.get(i), b = list.get(i + 1);
if((b - a) / 2 > maxLen){
result = (a + b) / 2;
maxLen = (b - a) / 2;
}
}
if(N - 1 - list.get(list.size() - 1) > maxLen){
result = N - 1;
}
//寻找第一个大于result的数,为了保证list是有序的(非递减的)。
int index = lowerBound(list, result);
list.add(index, result);
return result;
}
//lowerBound方法(list是非递减的):
//在list中找到第一个 大于等于 result 的那个数的索引。
//基本思想是:二分查找
//例如 list : {1,2,3,4,6,7,8} ; result : 5; 返回值是4.
//因为lsit中第一的大于等于5的数是6,其索引为4.
private int lowerBound(List<Integer> list, int result){
int min = 0, max = list.size();
while(min < max) {
int mid = (max + min) / 2;
if(result <= list.get(mid)) {
max = mid;
}else{
min = mid + 1;
}
}
return min;
}
//直接将p从list中移除就好了
public void leave(int p) {
list.remove(new Integer(p));
}
}
/**
* Your ExamRoom object will be instantiated and called as such:
* ExamRoom obj = new ExamRoom(N);
* int param_1 = obj.seat();
* obj.leave(p);
*/
做这道题花了我好长的时间,可能是第一次遇到对我来说这么难的题,所以说"革命尚未成功,同志任需努力!"
最后,如果看到的大佬有更优的解答,欢迎指导,谢谢哦!