在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