天天看点

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);
    }
};