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();
}
}