【贪心】leetcode630.课程表III

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

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

返回你最多可以修读的课程数目。
【贪心】leetcode630.课程表III_第1张图片
思路:
贪心+优先级队列
参考官方解答:https://leetcode-cn.com/problems/course-schedule-iii/solution/ke-cheng-biao-iii-by-leetcode-solution-yoyz/
【贪心】leetcode630.课程表III_第2张图片
【贪心】leetcode630.课程表III_第3张图片

解答:

class Solution:
    def scheduleCourse(self, courses: List[List[int]]) -> int:
        #贪心算法
        courses.sort(key=lambda c:c[1])
        #print(courses)
        q=[]
        #total:优先级队列中所有课程的总时间
        total=0

        for t,d in courses:
            if total+t<=d:
                total+=t
                #python默认是小根堆
                heapq.heappush(q,-t)
            elif q and -q[0]>t:
                total+=q[0]+t
                heapq.heappop(q)
                heapq.heappush(q,-t)

        return len(q)

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