Leetcode 797.所有可能的路径
1 题目描述(Leetcode题目链接)
给一个有 n 个结点的有向无环图,找到所有从 0 到 n-1 的路径并输出(不要求按顺序)
二维数组的第 i 个数组中的单元都表示有向图中 i 号结点所能到达的下一些结点(译者注:有向图是有方向的,即规定了a→b你就不能从b→a)空就是没有下一个结点了。
示例:
输入: [[1,2], [3], [3], []]
输出: [[0,1,3],[0,2,3]]
解释: 图是这样的:
0--->1
| |
v v
2--->3
这有两条路: 0 -> 1 -> 3 和 0 -> 2 -> 3.
提示:
- 结点的数量会在范围 [2, 15] 内。
- 你可以把路径以任意顺序输出,但在路径内的结点的顺序必须保证。
2 题解
标准的深度优先搜索。
class Solution:
def allPathsSourceTarget(self, graph: List[List[int]]) -> List[List[int]]:
n = len(graph)
retv = []
def dfs(i, path):
if path[-1] == n - 1:
retv.append(path)
return
for node in graph[i]:
dfs(node, path + [node])
dfs(0, [0])
return retv