LeetCode 797. All Paths From Source to Target
LeetCode 797. All Paths From Source to Target (所有可能的路径)
题目
链接
https://leetcode-cn.com/problems/all-paths-from-source-to-target/
问题描述
给你一个有 n 个节点的 有向无环图(DAG),请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序)
graph[i] 是一个从节点 i 可以访问的所有节点的列表(即从节点 i 到节点 graph[i][j]存在一条有向边)。
示例
输入:graph = [[1,2],[3],[3],[]]
输出:[[0,1,3],[0,2,3]]
解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3
提示
输入:graph = [[1,2],[3],[3],[]]
输出:[[0,1,3],[0,2,3]]
解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3
思路
考虑到有向无环图,就无需判断visit了,直接用图的遍历算法。
复杂度分析
时间复杂度 O(n*2^n)
空间复杂度 O(n)
代码
Java
List<List<Integer>> res = new LinkedList<>();
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
LinkedList<Integer> path = new LinkedList<>();
traverse(graph, 0, path);
return res;
}
void traverse(int[][] graph, int s, LinkedList<Integer> path) {
path.add(s);
int n = graph.length;
if (n - 1 == s) {
res.add(new LinkedList<>(path));
path.removeLast();
return;
}
for (int v : graph[s]) {
traverse(graph, v, path);
}
path.removeLast();
}