本文最后更新于 177 天前,其中的信息可能已经有所发展或是发生改变。
给你一个正整数 n
,它表示一个 有向无环图 中节点的数目,节点编号为 0
到 n - 1
(包括两者)。
给你一个二维整数数组 edges
,其中 edges[i] = [fromi, toi]
表示图中一条从 fromi
到 toi
的单向边。
请你返回一个数组 answer
,其中 answer[i]
是第 i
个节点的所有 祖先 ,这些祖先节点 升序 排序。
如果 u
通过一系列边,能够到达 v
,那么我们称节点 u
是节点 v
的 祖先 节点。
示例 :
输入:n = 8, edgeList = [[0,3],[0,4],[1,3],[2,4],[2,7],[3,5],[3,6],[3,7],[4,6]] 输出:[[],[],[],[0,1],[0,2],[0,1,3],[0,1,2,3,4],[0,1,2,3]] 解释: 上图为输入所对应的图。 - 节点 0 ,1 和 2 没有任何祖先。 - 节点 3 有 2 个祖先 0 和 1 。 - 节点 4 有 2 个祖先 0 和 2 。 - 节点 5 有 3 个祖先 0 ,1 和 3 。 - 节点 6 有 5 个祖先 0 ,1 ,2 ,3 和 4 。 - 节点 7 有 4 个祖先 0 ,1 ,2 和 3 。
提示:
1 <= n <= 1000
0 <= edges.length <= min(2000, n * (n - 1) / 2)
edges[i].length == 2
0 <= fromi, toi <= n - 1
fromi != toi
- 图中不会有重边。
- 图是 有向 且 无环 的。
解一(逆向 DFS)
class Solution:
def getAncestors(self, n, edges):
g = [[] for _ in range(n)]
for x, y in edges:
g[y].append(x) # 反向建图
def dfs(x):
vis[x] = True # 避免重复访问
for y in g[x]:
if not vis[y]:
dfs(y) # 只递归没有访问过的点
ans = [None] * n
for i in range(n):
vis = [False] * n
dfs(i) # 从 i 开始 DFS
vis[i] = False # ans[i] 不含 i
ans[i] = [j for j, b in enumerate(vis) if b]
return ans
解二:(正向 DFS)
class Solution:
def getAncestors(self, n, edges):
g = [[] for _ in range(n)]
for x, y in edges:
g[x].append(y)
def dfs(x):
vis[x] = start # 避免重复访问
for y in g[x]:
if vis[y] != start:
ans[y].append(start) # start 是访问到的点的祖先
dfs(y) # 只递归没有访问过的点
ans = [[] for _ in range(n)]
vis = [-1] * n
for start in range(n):
dfs(start) # 从 start 开始 DFS
return ans