题解 | #主持人调度(二)#

主持人调度(二)

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方法。最后堆的大小就是主持人个数。

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-26 15:46
已编辑
字节国际 电商后端 24k-35k
点赞 评论 收藏
分享
牛客963010790号:为什么还要收藏
点赞 评论 收藏
分享
威猛的小饼干正在背八股:挂到根本不想整理
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务