【LeetCode每日一题:855. 考场就座~~~有序表TreeSet】

最新推荐文章于 2025-11-23 20:02:12 发布

原创最新推荐文章于 2025-11-23 20:02:12 发布 · 307 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #Java #有序表 #TreeSet

LeetCode每日一题打卡专栏收录该内容

607 篇文章

订阅专栏

该博客讨论了一种实现策略，用于解决在考场中分配座位的问题，确保每个学生坐在与最近的人最大距离的位置。通过使用有序表（TreeSet）来存储座位，实现了`ExamRoom`类的`seat()`和`leave()`方法。`seat()`方法根据考场状态决定学生应坐在哪个座位，`leave()`方法则从有序表中移除离开的学生。示例展示了在不同场景下如何调用这些方法并返回正确的座位分配结果。

题目描述

在考场里，一排有 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 上。

求解思路

ExamRoom():方法中我们初始化座位数和有序表TreeSet。
seat():方法中的实现逻辑是这样的，首先我们判断如果考场里没有人，那么学生就坐在 0 号座位上。因为题目中给定的要求是:如果有多个这样的座位，他会坐在编号最小的座位上。所以说我们要有序表中最左位置开始逻辑判断，接下来然后再判断最右位置，我们取使得距离最近最大化的结果座位位置放入到有序表中，然后返回当前位置。

a和b俩点距离的中点公式:(b-a)/2
a和b俩点下标的中点公式:(b+a)/2
leave(int p):方法中的实现逻辑是我们直接从有序表中移除p位置的学生。

实现代码

class ExamRoom {
    
    private TreeSet<Integer> set;
    
    private int n;
    
    public ExamRoom(int n) {
        this.n = n;
        set = new TreeSet<>(); 
    }
    public int seat() {
        // 如果考场里没有人，那么学生就坐在 0 号座位上
        if (set.size() == 0) {
            set.add(0); 
            return 0;
        }
        // 先对最左开始判断
        // 初始化为选择最左的长度
        int pre = set.first(), ans = set.first(), idx = 0; 
        for (int x : set) {
            if (ans < (x - pre) / 2) {
                ans = (x - pre) / 2;
                idx = (x + pre) / 2;
            }
            pre = x;
        }
        // 然后再对最右进行判断
        int d = n - 1 - set.last();
        if (ans < d) {
            ans = d; 
            idx = n - 1;
        }
        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);
 */