207-210. Course Schedule [I & II] [Medium] 拓扑排序/课程顺序

最新推荐文章于 2021-07-24 13:49:53 发布

木木歹朱

最新推荐文章于 2021-07-24 13:49:53 发布

阅读量162

点赞数

CC 4.0 BY-SA版权

本文链接：https://blog.youkuaiyun.com/qq_35319215/article/details/98803527

本文深入探讨了课程调度问题，利用拓扑排序算法解决课程先修条件的冲突，确保所有课程可以顺利完成。通过记录课程入度并迭代寻找入度为零的课程，实现课程的有效排序。文章提供了详细的Python代码示例，包括判断课程是否可以完成（CourseSchedule）和生成课程完成顺序（CourseScheduleII）。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

207. Course Schedule

拓扑排序
先用一个数组indegree记录每个课程的入度，每次找一个入度为0的课程，把和该课程的边全部删掉，循环课程数次，如果找不到入度为0的点，则返回False，注意，当找到入度为0的点，需要更新，把它换成-1

class Solution(object):
    def canFinish(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: bool
        """
        indegree = [0] * numCourses
        for pre in prerequisites:
            indegree[pre[0]] += 1
        for i in range(numCourses):
            zero = -1
            for i in range(numCourses):
                if indegree[i] == 0:
                    zero = i
                    indegree[i] = -1
                    break
            if zero == -1:
                return False
            for pre in prerequisites:
                if pre[1] == zero:
                    indegree[pre[0]] -= 1
        return True

210. Course Schedule II

仍然一样的思路，只是需要把路径用一个数组记录下来。

class Solution(object):
    def findOrder(self, numCourses, prerequisites):
        """
        :type numCourses: int
        :type prerequisites: List[List[int]]
        :rtype: List[int]
        """
        res = []
        indegree = [0] * numCourses
        for pre in prerequisites:
            indegree[pre[0]] += 1
        for i in range(numCourses):
            zero = -1
            for i in range(numCourses):
                if indegree[i] == 0:
                    zero = i
                    indegree[i] = -1
                    break
            if zero == -1:
                return []
            res.append(zero)
            for pre in prerequisites:
                if pre[1] == zero:
                    indegree[pre[0]] -= 1
        return res