LeetCode刷题日记2021-12-14/630. 课程表 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 天完成它，这已经超出了关闭日期。
示例 2：

输入：courses = [[1,2]]
输出：1

示例 3：

输入：courses = [[3,2],[4,3]]
输出：0

提示:

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

题解思路

• 将课程中的结束时间以升序排列 如果已修课程+当前课程的时间 小于 当前课程的最晚时间我们将其压入堆中

• 当前课程塞满的时候 比较当前课程持续时间与堆顶课程持续时间 如果当前短 我们将其压入 堆顶弹出

  • 这样做为了 以更少的时间学习相同多的课程 同时保证后续可以学习更多课程 且不比原来的少

题解代码

class Solution:
	def scheduleCourse(self, courses: List[List[int]]) -> int:
		total,pq=0,[]
		courses.sort(key=lambda x:x[1])
		
		for time,end in courses:
			if time+total<=end:
				heapq.heappush(pq,-time)
				total+=time
			elif pq and -pq[0]>time:
				total-=-pq[0]-time
				heapq.heappop(pq)
				heapq,heappush(pq,-time)
		return len(pq)

• 我们将time元素以负数形式插入heap中 这样始终在维护一个最大堆