题解 | #主持人调度(二)#
主持人调度(二)
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方法。最后堆的大小就是主持人个数。