开启掘金成长之旅！这是我参与「掘金日新计划 · 12 月更文挑战」的第33天，点击查看活动详情

前言

小白算法比较菜，希望能激励我每日更新，从leetcode第一题开始，2022年目标2000分，现在1968！！

12月30日每日一题

855. 考场就座

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

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

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

示例 1：

输入：["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 上。

代码

这道题可以理解为男生进男厕所如何挑选小便池子的问题

max记录最大的距离，l和r分别记录最大距离时的左右边界

对于中间的判断，要除以2进行比较，例如0 4 9，距离分别为4和5，但应该插入2而不是6

遍历完之后，最后再判断n和set最后一个元素的距离是否最大

解法

class ExamRoom {
    TreeSet<Integer> set;
    int r;
    public ExamRoom(int n) {
        set = new TreeSet<>();
        r = n - 1;
    }

    public int seat() {
        if(set.isEmpty()) {
            set.add(0);
            return 0;
        }
        int pre = -1;
        int max = 0;
        int res = 0;
        for(int i : set) {
            if(pre == -1) {
                max = i;
            } else if((i - pre) / 2 > max) {
                max = (i - pre) / 2 ;
                res = i + pre >> 1;
            }
            pre = i;
        }
        if(r - pre > max) {
            res = r;
        }
        set.add(res);
        return res;
    }

    public void leave(int p) {
        set.remove(p);
    }
}

[杨小白]_leetcode_力扣_12月30日每日一题_855. 考场就座

前言

855. 考场就座

代码