在考场里,一排有 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 上。
思路:数据结构的考察,我们采用java中的treeset,由于内部自动排序,因此对于第一种操作我们用O(n1)的复杂度遍历整个集合找到使得学生之间距离最大的位置,并将相应位置插入集合;第二种操作我们直接使用Treeset中的remove函数操作即可,复杂度为O(logn2),其中n1和n2分别为两种操作的次数。
class ExamRoom {
int n;
TreeSet<Integer> st;
public ExamRoom(int N) {
n=N;
st=new TreeSet<>();
}
public int seat() {
int stu=0;
if(st.size()>0) {
int dist=st.first();
Integer pre=null;
for(Integer now : st) {
if(pre!=null) {
int d=(now-pre)/2;
if(d>dist) {
dist=d;
stu=pre+d;
}
}
pre=now;
}
if(n-1-st.last()>dist)
stu=n-1;
}
st.add(stu);
return stu;
}
public void leave(int p) {
st.remove(p);
}
}