题解 | #主持人调度(二)#
主持人调度(二)
https://www.nowcoder.com/practice/4edf6e6d01554870a12f218c94e8a299
import java.util.*;
public class Solution {
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
* 计算成功举办活动需要多少名主持人
* @param n int整型 有n个活动
* @param startEnd int整型二维数组 startEnd[i][0]用于表示第i个活动的开始时间,startEnd[i][1]表示第i个活动的结束时间
* @return int整型
*/
public int minmumNumberOfHost (int n, int[][] startEnd) {
// write code here
if(n <= 1){
return n;
}
// 将活动按照开始时间排序,由于是数组的形式,因此可以用Arrays的sort方法
Arrays.sort(startEnd,new Comparator<int []>(){
@Override
public int compare(int[] interval1, int [] interval2){
return Integer.compare(interval1[0],interval2[0]);
}
});
// 用最小堆来记录活动的结束时间
PriorityQueue<Integer> minHeap = new PriorityQueue<>();
// 先把第一个活动结束时间添加进去
// offer方法就是往优先级队列中插入元素
minHeap.offer(startEnd[0][1]);
for(int i = 1; i< startEnd.length;i++){
int[] interval = startEnd[i];
// 如果当前活动开始时间比堆里面记录的最早结束时间要晚,不用增加主持人,但需要改变该主持人活动结束时间
// peek返回优先级队列的头部元素,也就是最小的元素
if(interval[0] >= minHeap.peek()){
// poll方法执行移除队列头部元素操作,也就是最小的元素
minHeap.poll();
}
minHeap.offer(interval[1]);
}
return minHeap.size();
}
}
这个题目用到了我不熟悉的数据结构,优先级队列,这里面用的是最小堆,是一棵完全二叉树(除了最后一层,其他层都是满的)。
之前我的想法是:跟区间合并一样,合并一次就增加一个主持人,但这种想法不行。例如都是a b a b a b 的情况,合并计算的主持人个数就为区间个数-1,实际上一个人可以负责a和b,因此这种做法不行。
堆的思路就是,也先对活动按照开始时间排序,然后在堆里面记录活动的结束时间。每次遍历,都把活动开始的时间与最早结束的那个时间比较,如果可以接上,就让这个主持人继续,要使用poll方法把这个最早结束时间顶出来。如果接不上,直接执行offer方法。最后堆的大小就是主持人个数。
凡岛公司福利 757人发布