课程表2

在DFS中,需用visit来记录顶点的访问情况,避免无限递归,这点是没有异议的,同时为了满足能判断环这个条件,用额外的onCycle来记录从某个点开始递归时的路径,这里由于并不需要关注路径,onCycle直接存True和False表示状态就好。
采用回溯的思想,当一条递归线走完之后,将onCycle状态再恢复成False。
这里给DFS增加了返回值,如果有环直接返回False,一旦出现环,从所有嵌套递归中立刻返回。
拓扑排序实际上就是后序遍历的逆序输出,所以也可以在每个顶点该输出的地方,将其插入在结果集的第一个位置res.insert(0,v),这样最后的结果可以不用逆序。

拓扑排序

class Solution:
    def findOrder(self, numCourses: int, prerequisites: List[List[int]]) -> List[int]:
        G=[[] for _ in range(numCourses)]
        for i,j in prerequisites:
            G[j].append(i)
        visit=[False]*numCourses
        onCycle=[False]*numCourses
        res=[]
        def dfs(v):
            visit[v]=True
            onCycle[v]=True
            for w in G[v]:
                if not visit[w]:
                    if not dfs(w):
                        return False
                elif onCycle[w]:
                    return False
            onCycle[v]=False
            res.insert(0,v)
            return True
        for v in range(numCourses):
            if not visit[v]:
                if not dfs(v):
                    return []
        return res

课程表2_第1张图片

你可能感兴趣的:(打卡)