LeetCode207、210 课程表（图 dfs 拓扑排序）

可keke

已于 2024-04-02 17:24:39 修改

阅读量255

点赞数 2

分类专栏：解题笔记文章标签：深度优先算法 python

于 2024-04-02 15:46:11 首次发布

本文链接：https://blog.csdn.net/qq_44722189/article/details/137272908

版权

解题笔记专栏收录该内容

10 篇文章 0 订阅

订阅专栏

文章讲述了两种解决方案，分别是`classSolution`中的深度优先搜索方法对课程表进行拓扑排序，以及`findOrder`版本，确保先决课程完成后添加到结果中。作者展示了如何利用依赖关系图进行遍历，以判断是否存在环并确定课程顺序。

摘要由CSDN通过智能技术生成

官方题解主要思想是使用dfs进行拓扑排序。

课程表

class Solution:
    def canFinish(self, numCourses: int, prerequisites: List[List[int]]) -> bool:
        edges = collections.defaultdict(list) 
        visited = [0] * numCourses # 将列表初始化为numCourses个0
        # result = []
        valid = True

        for info in prerequisites:
            edges[info[1]].append(info[0]) # info[1]是哪些课程的依赖（先决条件）    
                
        def dfs(u: int):
            nonlocal valid
            visited[u] = 1
            for v in edges[u]:
                if visited[v] == 0:
                    dfs(v)
                    if not valid:
                        return
                elif visited[v] == 1:
                    valid = False
                    return
            visited[u] = 2
            # result.append(u)
        
        for i in range(numCourses):
            if valid and not visited[i]:
                dfs(i)
        
        return valid

课程表二

class Solution:
    def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
        edges = collections.defaultdict(list)
        visited = [0]*numCourses
        result = []
        valid =True

        for info in prerequisites:
            edges[info[1]].append(info[0])

        def dfs(u):
            nonlocal valid
            visited[u]=1
            for v in edges[u]:
                if visited[v]==0:
                    dfs(v)
                elif visited[v]==1:
                    valid=False
                    return 

            visited[u]=2
            result.append(u)

        for i in range(numCourses):
            if valid and not visited[i]:
                dfs(i)

        if valid==True:
            result.reverse()
            return result
        else:
            return []