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 上。

方法1:
主要思路:
(1)使用set来对已经插入的位置进行保存;
(2)当插入新的位置时,遍历整个set中的各个可能的区间,并找出最大的且最靠前的区间;
(3)处理两端的特殊情形;
(4)删除时,直接在set中erase掉对应的元素即可;

class ExamRoom {
    private:
    int n;
    set<int> st;
public:
    ExamRoom(int N):n(N) {

    }
    
    int seat() {
    	//处理特殊的情形
        if(st.empty()){
            st.insert(0);
            return 0;
        }
        //标识区间
        int left=0;
        int right=0;
        int len=0;
        int pre=n;
        //找出合适的区间
        for(auto&it:st){
            if((it-pre)/2>len){
                len=(it-pre)/2;
                left=pre;
                right=it;
            }
            pre=it;
        }
        //处理两端的情形
        if(*st.begin()!=0&&*st.begin()>=len){
            len=*st.begin();
            left=0;
            right=0;
        }
        if(*st.rbegin()!=n-1&&n-1-*st.rbegin()>len){
            len=n-1-*st.rbegin();
            left=n-1;
            right=n-1;
        }
		//返回结果值
        st.insert(left+(right-left)/2);
        return left+(right-left)/2;
    }
    //删除对应的元素
    void leave(int p) { 
        st.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);
 */
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值