题目
给你一个有 n 个节点的 有向无环图(DAG),请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序)
二维数组的第 i 个数组中的单元都表示有向图中 i 号节点所能到达的下一些节点,空就是没有下一个结点了。
译者注:有向图是有方向的,即规定了 a→b 你就不能从 b→a 。
示例 1:
输入:graph = [[1,2],[3],[3],[]]
输出:[[0,1,3],[0,2,3]]
解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3
示例 2:
输入:graph = [[4,3,1],[3,2,4],[3],[4],[]]
输出:[[0,4],[0,3,4],[0,1,3,4],[0,1,2,3,4],[0,1,4]]
示例 3:
输入:graph = [[1],[]]
输出:[[0,1]]
示例 4:
输入:graph = [[1,2,3],[2],[3],[]]
输出:[[0,1,2,3],[0,2,3],[0,3]]
示例 5:
输入:graph = [[1,3],[2],[3],[]]
输出:[[0,1,2,3],[0,3]]
提示:
- n == graph.length
- 2 <= n <= 15
- 0 <= graph[i][j] < n
- graph[i][j] != i(即,不存在自环)
- graph[i] 中的所有元素 互不相同
- 保证输入为 有向无环图(DAG)
题解
方法一:深度搜索DFS
一看到这种找所有可能的路径,并且是有向无环图,就想到了使用DFS。
自己写的:
class Solution {
List<List<Integer>> res = null;
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
int n = graph.length;
res = new ArrayList<>();
boolean[] visited = new boolean[n];
List<Integer> path = new ArrayList<Integer>();
path.add(0);
bfs(graph, visited, 0, path, n);
return res;
}
public void bfs(int[][] graph, boolean[] visited, int cur, List<Integer> path, int n){
if(visited[cur] == true) return;
if(cur == n - 1){
res.add(path);
return;
}
visited[cur] = true;
for(int next : graph[cur]){
List<Integer> newPath = new ArrayList(path); // 深复制之前的路径
newPath.add(next); // 添加下一个结点
bfs(graph, visited, next, newPath, n);
}
visited[cur] = false;
}
}
后面发现由于是有向无环图,其实不需要使用visited来标记是否访问,因为一条路径的搜索过程是不会访问到同一个结点的。
修改后:
class Solution {
List<List<Integer>> res = null;
int n;
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
n = graph.length;
res = new ArrayList<>();
List<Integer> path = new ArrayList<Integer>();
path.add(0);
bfs(graph, 0, path);
return res;
}
public void bfs(int[][] graph, int cur, List<Integer> path){
if(cur == n - 1){
res.add(new ArrayList(path));
return;
}
for(int next : graph[cur]){
path.add(next); // 添加下一个结点
bfs(graph, next, path);
path.remove(path.size() - 1); // 还原
}
}
}