Leetcode.855考场就坐

题目链接

Leetcode.855 考场就坐

题目描述

在考场里，一排有 N 个座位，分别编号为 0, 1, 2, …, N-1 。

当学生进入考场后，他必须坐在能够使他与离他最近的人之间的距离达到最大化的座位上。如果有多个这样的座位，他会坐在编号最小的座位上。(另外，如果考场里没有人，那么学生就坐在 0 号座位上。)

返回 ExamRoom(int N) 类，它有两个公开的函数：其中，函数 ExamRoom.seat() 会返回一个 int （整型数据），代表学生坐的位置；函数 ExamRoom.leave(int p) 代表坐在座位 p 上的学生现在离开了考场。每次调用 ExamRoom.leave§ 时都保证有学生坐在座位 p 上。

示例：

输入：[“ExamRoom”,“seat”,“seat”,“seat”,“seat”,“leave”,“seat”],
[[10],[],[],[],[],[4],[]]
输出：[null,0,9,4,2,null,5]
解释： ExamRoom(10) ->null seat() -> 0，没有人在考场里，那么学生坐在 0 号座位上。
seat() -> 9，学生最后坐在 9 号座位上。
seat() -> 4，学生最后坐在 4 号座位上。
seat() -> 2，学生最后坐在 2 号座位上。
leave(4) -> null
seat() -> 5，学生最后坐在 5 号座位上。

提示：

1 <= N <= $10^9$
在所有的测试样例中 ExamRoom.seat() 和 ExamRoom.leave() 最多被调用$10^4$ 次。
保证在调用 ExamRoom.leave( p ) 时有学生正坐在座位 p 上。

分析：

对于本题又可以理解为，男生小便的时候如何最大程度的避免尴尬。
对于一个 [ L , R ]的区间，可以很容易的得出 选取中点 （L + R ）/ 2 就可以使距离最近的点的距离最大化，最大距离为 (R - L) / 2。
例如：
[0,9] 选取中点 4 , 即 [0 , 4 , 9] 距离最近的点的距离为 4
[1,5] 选取中点3，即 [1 , 3 , 5] 距离最近的点的距离为2

seat():

我们用一个有序集合 set 来记录已经选择的位置。
初始时 set 为空，就直接把 0 加入到集合中，直接返回0。
set 不为空的情况下，就要遍历 set 选出一个距离最大的中点出来，在这个过程中需要特判 选择 0 和 n - 1的情况。
对于这两个特殊的点来说：
0 点的距离为 set 的第一个元素 - 0，即 set.first()
n - 1 点的距离为 n - 1 - set 的最后一个元素，即 n - 1 - set.last();

leave():

leave 直接利用 set 自带的删除函数即可。

• 时间复杂度：seat(): $O(n)$ leave() : $O(logn)$

cpp代码：

class ExamRoom {
public:
    set<int> S;
    int n;
    ExamRoom(int n) {
        this->n = n;
    }
    
    int seat() {
        if(S.empty()){
            S.insert(0);
            return 0;
        }
        auto begin = S.begin();
        int ans = *begin,pre = *begin;
        int idx = 0;
        for(auto &x:S){
            if(ans < (x - pre)/2){
                ans = (x - pre)/2;
                idx = (pre + x)/2;
            }
            pre = x;
        }
        auto it = S.end();
        it--;
        if(n - 1 - *it > ans){
            idx = n - 1;
        }

        S.insert(idx);
        return idx;
    }
    
    void leave(int p) {
        S.erase(p);
    }
};

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

Java代码：

class ExamRoom {
    //记录选择的位置
    TreeSet<Integer> set;
    int n;
    public ExamRoom(int n) {
        set = new TreeSet<>();
        this.n = n;
    }
    
    public int seat() {
        //如果set为空 直接选择 0 位置，返回
        if(set.size() == 0){
            set.add(0);
            return 0;
        }

        //遍历 更新d为最大的距离，idx 为最大距离的那个位置
        int idx = 0;
        //pre 指的是前一个位置
        int d = set.first(),pre = set.first();
        for(int x:set){
            if((x - pre)/2 > d){
                d = (x - pre)/2;
                idx = (x + pre)/2;
            }
            //更新pre
            pre = x;
        }

        //判断最后一个点 n - 1 的距离 是否大于 d 
        if(n - 1 - set.last() > d){
            idx = n - 1;
        }

        //将选择好的位置 idx 记录到set中
        set.add(idx);
        return idx;
    }
    
    public void leave(int p) {
        
        set.remove(p);
    }
}

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