leetcode855. 考场就座

题目地址

https://leetcode-cn.com/problems/exam-room/

题目描述

在考场里，一排有 N 个座位，分别编号为 0, 1, 2, ..., N-1 。
当学生进入考场后，他必须坐在能够使他与离他最近的人之间的距离达到最大化的座位上。如果有多个这样的座位，他会坐在编号最小的座位上。(另外，如果考场里没有人，那么学生就坐在 0 号座位上。)
返回 ExamRoom(int N) 类，它有两个公开的函数：其中，函数 ExamRoom.seat() 会返回一个 int （整型数据），代表学生坐的位置；函数 ExamRoom.leave(int p) 代表坐在座位 p 上的学生现在离开了考场。每次调用 ExamRoom.leave(p) 时都保证有学生坐在座位 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 上。

代码实现

javascript

/**
 * @param {number} n
 */
var ExamRoom = function(n) {
    this.n = n;
    // 用students来存储已经有人的座位
    this.students = [];
};

/**
 * @return {number}
 */
ExamRoom.prototype.seat = function() {
    let student = 0;
    if (this.students.length > 0) {
        let dist = this.students[0];
        let prev = dist;
        for (let i = 1; i < this.students.length; i++) {
            let d = Math.floor((this.students[i]-prev)/2);
            if (d > dist) {
                dist = d;
                student = prev + dist;
            }
            prev = this.students[i];
        }    
        if (this.n - 1 - this.students[this.students.length-1] > dist) student = this.n - 1;
    }

    if (!this.students.includes(student)) this.students.push(student);
    this.students.sort((a,b) => a-b);
    return student;
};

/** 
 * @param {number} p
 * @return {void}
 */
ExamRoom.prototype.leave = function(p) {
    this.students = this.students.filter(v => v!==p);
};