算法日常——找到最终的安全状态

Posted by Maolong on Thursday, August 5, 2021

问题:

有一个有 n 个节点的有向图,节点按 0n - 1 编号。图由一个 索引从 0 开始 的 2D 整数数组 graph表示, graph[i]是与节点 i 相邻的节点的整数数组,这意味着从节点 i 到 graph[i]中的每个节点都有一条边。

如果一个节点没有连出的有向边,则它是 终端节点 。如果没有出边,则节点为终端节点。如果从该节点开始的所有可能路径都通向 终端节点 ,则该节点为 安全节点

返回一个由图中所有 安全节点 组成的数组作为答案。答案数组中的元素应当按 升序 排列。

 

示例 1:

Illustration of graph

输入:graph = [[1,2],[2,3],[5],[0],[5],[],[]]
输出:[2,4,5,6]
解释:示意图如上。
节点 5 和节点 6 是终端节点,因为它们都没有出边。
从节点 2、4、5 和 6 开始的所有路径都指向节点 5 或 6 。

示例 2:

输入:graph = [[1,2,3,4],[1,2],[3,4],[0,4],[]]
输出:[4]
解释:
只有节点 4 是终端节点,从节点 4 开始的所有路径都通向节点 4 。

 

提示:

  • n == graph.length
  • 1 <= n <= 104
  • 0 <= graph[i].length <= n
  • 0 <= graph[i][j] <= n - 1
  • graph[i] 按严格递增顺序排列。
  • 图中可能包含自环。
  • 图中边的数目在范围 [1, 4 * 104] 内。
Related Topics
  • 深度优先搜索
  • 广度优先搜索
  • 拓扑排序

  • 👍 354
  • 👎 0
  • 题解:

    // @lc code=start
    func eventualSafeNodes(graph [][]int) []int {
    	status := make(map[int]int)
    	for i := range graph {
    		mark(&graph, i, &status)
    	}
    	res := make([]int, 0)
    	for k, v := range status {
    		if v == 1 {
    			res = append(res, k)
    		}
    	}
    	sort.Ints(res)
    	return res
    }
    
    func mark(graph *[][]int, i int, status *map[int]int) int {
    	if _, ok := (*status)[i]; !ok {
    		(*status)[i] = 0
    	} else {
    		return (*status)[i]
    	}
    	if len((*graph)[i]) == 0 {
    		(*status)[i] = 1
    		return (*status)[i]
    	}
    	for _, v := range (*graph)[i] {
    		if v == i {
    			(*status)[i] = -1
    			continue
    		}
    		if _, ok := (*status)[v]; ok && (*status)[v] == 0 {
    			(*status)[i] = -1
    			return (*status)[i]
    		}
    		s := mark(graph, v, status)
    		if s == -1 {
    			(*status)[i] = -1
    			return (*status)[i]
    		}
    	}
    	if (*status)[i] == 0 {
    		(*status)[i] = 1
    	}
    	return (*status)[i]
    }
    
    // @lc code=end
    

    成绩:

    • 112112 cases passed (124 ms)
    • Your runtime beats 37.5 % of golang submissions
    • Your memory usage beats 97.92 % of golang submissions (7.4 MB)

    「真诚赞赏,手留余香」

    Maolong’s Blog

    真诚赞赏,手留余香

    使用微信扫描二维码完成支付