
思路:
- 使用set(有序集合)记录seat的位置。
- 如果set为空,放入索引为0的位置。
- 遍历set,记录两个人之间距离的一半(pre+这个距离就是中点),如果这个距离比最大距离大,则记录pos和更新最大距离。因为1/2=0。所以将maxDist的初始值设为0。pre设为-1(表示第一元素钱前面没有元素)。pre表示cur前面的那个元素。
- 最后判断如果放在n-1的位置上,距离是不是会大(n-1-pre>maxDsit),如果大就更新pos。
- 得到pos后,将pos插入set,返回pos。
class ExamRoom {
public:
set<int> s;
int n;
ExamRoom(int N) {
this->n=N;
}
int seat() {
if(s.size()==0) {
s.insert(0);
return 0;
}//set为空,直接插入
int pos=0,pre=-1,maxDist=0;
for(auto cur:s){
int dist=(cur-pre)/2;//pre+dist就是元素应该放的位置
if(dist>maxDist){//有更大的距离
pos=pre==-1?0:pre+dist; //如果能进来,表示cur!=0,因为(0-(-1))/2=0;
maxDist=dist;//更新dist
}
pre=cur;
}
if(n-pre-1>maxDist) pos=n-1;//最后考虑n-1位置的情况
s.insert(pos);
return pos;
}
void leave(int p) {
s.erase(p);
}
};