其他分享
首页 > 其他分享> > LeetCode-802. 找到最终的安全状态

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