题⽬:假设有⼀个考场,考场有⼀排共 N 个座位,索引分别是 [0…N-1] ,考⽣会陆续进⼊考场考试,并且可能在任何时候离开考场。
你作为考官,要安排考⽣们的座位,满⾜:每当⼀个学⽣进⼊时,你需要最⼤化他和最近其他⼈的距离;如果有多个这样的座位,安排到他到索引最⼩的那个座位。
也就是请你实现下⾯这样⼀个类:
class ExamRoom {
// 构造函数,传⼊座位总数 N
public ExamRoom(int N);
// 来了⼀名考⽣,返回你给他分配的座位
public int seat();
// 坐在 p 位置的考⽣离开了
// 可以认为 p 位置⼀定坐有考⽣
public void leave(int p);
}
1. 抽象出题目本质
举5个座位的小例子,先不要考虑leave,抽象出题目本质:
如果将每两个相邻的考⽣看做线段的两端点,新安排考⽣就是找最⻓的线段,然后让该考⽣在中间把这个线段「⼆分」,中点就是给他分配的座位。 leave§ 其实就是去除端点 p ,使得相邻两个线段合并为⼀个。
2. 思路分析
根据上述思路,⾸先需要把坐在教室的学⽣抽象成线段,我们可以简单的⽤⼀个⼤⼩为 2 的数组表⽰。
另外,需要找到「最⻓
」的线段,还需要去除线段,增加线段。
但凡遇到在动态过程中取最值的要求,肯定要使⽤有序数据结构,我们常⽤的数据结构就是⼆叉堆和平衡⼆叉搜索树了。
⼆叉堆实现的优先级队列取最值的时间复杂度是 O(logN),但是只能删除最⼤值。
平衡⼆叉树也可以取最值,也可以修改、删除任意⼀个值,⽽且时间复杂度都是 O(logN)。
综上,⼆叉堆不能满⾜ leave 操作,应该使⽤平衡⼆叉树。所以这⾥我们会⽤到 Java 的⼀种数据结构 TreeSet ,这是⼀种有序数据结构,底层由红⿊树维护有序性
3. 简化问题
⾸先,如果有多个可选座位,需要选择索引最⼩的座位对吧?我们先简化⼀下问题,暂时不管这个要求,实现上述思路。
编程技巧:使⽤⼀个「虚拟线段」让算法正确启动,这就和链表相关的算法需要「虚拟头结点」⼀个道理。
「虚拟线段」其实就是为了将所有座位表⽰为⼀个线段:
构造基本函数
(先不写具体的函数内容,只要函数名,来理清思路和问题结构,记得写注释)
// 将端点 p 映射到以 p 为左端点的线段
private Map<Integer, int[]> startMap;
// 将端点 p 映射到以 p 为右端点的线段
private Map<Integer, int[]> endMap;
// 根据线段⻓度从⼩到⼤存放所有线段
private TreeSet<int[]> pq;
private int N;
public ExamRoom(int N) {
this.N = N;
startMap = new HashMap<>(