力扣题目——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(); } }
下一篇:
数据结构与算法之计数排序