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;
    }
}
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章