【每日一题】885. 考场就坐

最新推荐文章于 2024-04-29 08:35:00 发布

昔我往矣wood

最新推荐文章于 2024-04-29 08:35:00 发布

阅读量951

点赞数

分类专栏： Leetcode 文章标签： leetcode 算法 c++ 数据结构

本文链接：https://blog.csdn.net/zhuge2017302307/article/details/128493871

版权

Leetcode 专栏收录该内容

47 篇文章 1 订阅

订阅专栏

重点：

（1）set的使用：

——*seats.begin()，*seats.end()；

——遍历：for(int cur:seats)或者auto；

855. 考场就座

难度中等171

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

解析：

寻找最优位置。扫描整个set即可，找出间隔最大的，把人放进去就行。时间空间复杂度都为O(n)。

代码：

class ExamRoom {
private:
    int n;
    set<int> seats;
    
public:
    ExamRoom(int n): n(n) {
    }
    
    int seat() {
        if(seats.empty()){ //当前为空，则坐0位置
            seats.insert(0);
            return 0;
        }
        int pos=0, pre=-1, maxdis=*seats.begin();
        //扫描set，寻找出最优位置，同时判断第一个和最后一个位置
        for(int cur:seats){
            if(pre!=-1){
                int d=(cur-pre)/2;
                if(d>maxdis){
                    maxdis=d;
                    pos=pre+d;
                }
            }
            pre=cur;
        }
        if(n-1-pre>maxdis)
            pos=n-1;
        seats.insert(pos);
        return pos;
    }
    void leave(int p) {
        seats.erase(p);
    }
};