题目描述:
在考场里,一排有 N 个座位,分别编号为 0, 1, 2, …, N-1 。
当学生进入考场后,他必须坐在能够使他与离他最近的人之间的距离达到最大化的座位上。如果有多个这样的座位,他会坐在编号最小的座位上。(另外,如果考场里没有人,那么学生就坐在 0 号座位上。)
返回 ExamRoom(int N) 类,它有两个公开的函数:其中,函数 ExamRoom.seat() 会返回一个 int (整型数据),代表学生坐的位置;函数 ExamRoom.leave(int p) 代表坐在座位 p 上的学生现在离开了考场。每次调用 ExamRoom.leave§ 时都保证有学生坐在座位 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 上。
方法1:
主要思路:
(1)使用set来对已经插入的位置进行保存;
(2)当插入新的位置时,遍历整个set中的各个可能的区间,并找出最大的且最靠前的区间;
(3)处理两端的特殊情形;
(4)删除时,直接在set中erase掉对应的元素即可;
class ExamRoom {
private:
int n;
set<int> st;
public:
ExamRoom(int N):n(N) {
}
int seat() {
//处理特殊的情形
if(st.empty()){
st.insert(0);
return 0;
}
//标识区间
int left=0;
int right=0;
int len=0;
int pre=n;
//找出合适的区间
for(auto&it:st){
if((it-pre)/2>len){
len=(it-pre)/2;
left=pre;
right=it;
}
pre=it;
}
//处理两端的情形
if(*st.begin()!=0&&*st.begin()>=len){
len=*st.begin();
left=0;
right=0;
}
if(*st.rbegin()!=n-1&&n-1-*st.rbegin()>len){
len=n-1-*st.rbegin();
left=n-1;
right=n-1;
}
//返回结果值
st.insert(left+(right-left)/2);
return left+(right-left)/2;
}
//删除对应的元素
void leave(int p) {
st.erase(p);
}
};
/**
* Your ExamRoom object will be instantiated and called as such:
* ExamRoom* obj = new ExamRoom(N);
* int param_1 = obj->seat();
* obj->leave(p);
*/