
思路:
- 使用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);
}
};