LeetCode 855. Exam Room

Posted 2021-01-21 約束の空

用set来做，seat() O(n)，leave() O(logn)

注意一些边界情况和下标即可，最容易遗漏的就是最左边和最右边的位置。

class ExamRoom {
public:
    ExamRoom(int N):n(N){};
    
    int seat() {
        if (s.size()==0){ s.insert(0); return 0;}
        int res=0;
        int max_dis=*s.begin();
        int prev=-1;
        
        for (int x:s){
            if (prev!=-1){
                int cur_dis = (x-prev)/2;
                if (cur_dis>max_dis){
                    max_dis = cur_dis;
                    res = prev+cur_dis;
                }
            }
            prev = x;
        }
        if (n-1-prev>max_dis){
            max_dis = n-1-prev;
            res = n-1;
        }
        s.insert(res);
        return res;
    }
    
    void leave(int p) {
        s.erase(p);
    }
    
private:
    int n;          // number of seats
    set<int> s;     // set records the seated seats
};

/**
 * Your ExamRoom object will be instantiated and called as such:
 * ExamRoom obj = new ExamRoom(N);
 * int param_1 = obj.seat();
 * obj.leave(p);
 */