力扣题目——630. 课程表 III(优先队列+贪心)

630. 课程表 III

力扣题目链接:https://leetcode.cn/problems/course-schedule-iii/

题意
这里有 n 门不同的在线课程,按从 1 到 n 编号。给你一个数组 courses ,其中 courses[i] = [durationi, lastDayi] 表示第 i 门课将会 持续 上 durationi 天课,并且必须在不晚于 lastDayi 的时候完成。

你的学期从第 1 天开始。且不能同时修读两门及两门以上的课程。

返回你最多可以修读的课程数目。

示例 1

输入:courses = [[100, 200], [200, 1300], [1000, 1250], [2000, 3200]]
输出:3
解释:
这里一共有 4 门课程,但是你最多可以修 3 门:
首先,修第 1 门课,耗费 100 天,在第 100 天完成,在第 101 天开始下门课。
第二,修第 3 门课,耗费 1000 天,在第 1100 天完成,在第 1101 天开始下门课程。
第三,修第 2 门课,耗时 200 天,在第 1300 天完成。
第 4 门课现在不能修,因为将会在第 3300 天完成它,这已经超出了关闭日期。

提示

  • 1 <= courses.length <= 104
  • 1 <= durationi, lastDayi <= 104

解题思路
按结束时间排序,越早结束的课程越早学,同时用优先队列(大顶堆)记录下符合条件(学习结束时间之前能学完)的课程。若是课程不符合条件,我们比较已经学的课程中耗时最长的课程和当前课程时长,若当前课程时长较长则不学,而当前课程时长较短则替换,保证已经学习课程时间总和一直是最短的,最后优先队列里存在的数据数量为最大所有课程数。

Java代码

class Solution {
    public int scheduleCourse(int[][] courses) {
        //按结束时间排序
        Arrays.sort(courses,(a, b) -> a[1] -b[1]);
        //大顶堆,保存学习课程的时长
        PriorityQueue<Integer> q = new PriorityQueue<>((a,b) -> b - a);
        //记录已经学习课程所用时间
        int time = 0;
        //遍历所有课程
        for (int[] course : courses) {
            int during = course[0];
            int end = course[1];
            //若可以学习,则加入优先队列
            if(time + during <= end){
                time += during;
                q.offer(during);
            }else if(!q.isEmpty() && q.peek() > during){ 
                //学习课程时间不够时,若队列不空,且当前课程时长相较前面所学最长课时短时,则学习较短时长的课程
                time = time - q.poll() + during;
                q.offer(during);
            }
        }
        return q.size();
    }
}

你可能感兴趣的:(leetcode,算法,贪心算法)