天天看點

leetcode-855. 考場就座

leetcode-855. 考場就座

思路:

  • 使用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);
    }
};