LeetCode-802. 找到最终的安全状态
作者:互联网
在有向图中,以某个节点为起始节点,从该点出发,每一步沿着图中的一条有向边行走。如果到达的节点是终点(即它没有连出的有向边),则停止。
对于一个起始节点,如果从该节点出发,无论每一步选择沿哪条有向边行走,最后必然在有限步内到达终点,则将该起始节点称作是 安全 的。
返回一个由图中所有安全的起始节点组成的数组作为答案。答案数组中的元素应当按 升序 排列。
该有向图有 n 个节点,按 0 到 n - 1 编号,其中 n 是 graph 的节点数。图以下述形式给出:graph[i] 是编号 j 节点的一个列表,满足 (i, j) 是图的一条有向边。
思路在代码中给出,使用了深度优先搜索,还有一种拓扑排序,没想明白
class Solution:
def eventualSafeNodes(self, graph: List[List[int]]) -> List[int]:
# 用0表示未被访问过,用1表示被访问过,用2表示安全
# 初始化为0,都没有被访问过
safe = [0] * len(graph)
def dnf(start) -> bool:
# 如果一个点被访问过,那么通过和2比较来判断是否已经被访问,或者说是否安全
if safe[start] > 0:
return safe[start] == 2
# 如果没有被访问过,那么设置为1表示被访问过
safe[start] = 1
# 深度遍历未被访问过的点,只要有一个结点返回False,那么就说明被访问过,存在环,不安全,返回False
for k in graph[start]:
if not dnf(k):
return False
# 如果之前都没有被访问过,那么该结点变为安全结点,返回True
safe[start] = 2
return True
return [i for i in range(len(graph)) if dnf(i)]
标签:return,start,找到,graph,safe,节点,访问,802,LeetCode 来源: https://blog.csdn.net/qq_41542989/article/details/119411256