MarcusV(南墙大佬的搬运工)

南墙大佬的CSDN博客地址

导航

Java实现 LeetCode 802 找到最终的安全状态 (DFS)

802. 找到最终的安全状态

在有向图中, 我们从某个节点和每个转向处开始, 沿着图的有向边走。 如果我们到达的节点是终点 (即它没有连出的有向边), 我们停止。

现在, 如果我们最后能走到终点,那么我们的起始节点是最终安全的。 更具体地说, 存在一个自然数 K, 无论选择从哪里开始行走, 我们走了不到 K 步后必能停止在一个终点。

哪些节点最终是安全的? 结果返回一个有序的数组。

该有向图有 N 个节点,标签为 0, 1, …, N-1, 其中 N 是 graph 的节点数. 图以以下的形式给出: graph[i] 是节点 j 的一个列表,满足 (i, j) 是图的一条有向边。

示例:
输入:graph = [[1,2],[2,3],[5],[0],[5],[],[]]
输出:[2,4,5,6]
这里是上图的示意图。

Illustration of graph

提示:

graph 节点数不超过 10000.
图的边数不会超过 32000.
每个 graph[i] 被排序为不同的整数列表, 在区间 [0, graph.length - 1] 中选取。

class Solution {
   public List<Integer> eventualSafeNodes(int[][] graph) {
        int len = graph.length;
        List<Integer> ans = new ArrayList<>(len);
        int[] counter = new int[len];
        for (int i=0; i<len; i++) {
            if (dfs(graph, counter, i))
                ans.add(i);
        }
        return ans;
    }
    
    private boolean dfs(int[][] graph, int[] counter, int curr) {
        if (counter[curr] > 0)
            return counter[curr] == 2;
        
        counter[curr] = 1;
        for (int child : graph[curr]) {
            if (counter[child] == 2) continue;
            if (counter[child]==1 || !dfs(graph, counter, child))
                return false;
        }
        counter[curr] = 2;
        return true;
    }
}

posted on 2020-05-08 12:38  MarcusV  阅读(52)  评论(0编辑  收藏  举报