给一个有 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]
内。 - 你可以把路径以任意顺序输出,但在路径内的结点的顺序必须保证。
本题要求一条从0到n-1的路径,图的点就是0—n-1,图的边通过二维数组来确定,方法就是回溯,定义一个变量表示当前搜索到的结点,再通过这个变量来搜索他的下一个结点。搜索完就删除这个结点,即回溯操作。若变量的值等于n-1时表示找到了一条从0—n-1的路径。定义一个临时列表来记录遍历到的结点,若找到了一个答案便将临时列表存入结果列表中。代码如下:
class Solution {
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
List<List<Integer>> res = new ArrayList<>();
ArrayList<Integer> temp = new ArrayList<>();
dfs(res, temp, 0, graph.length - 1, graph);
return res;
}
public void dfs(List<List<Integer>> res, ArrayList<Integer> a, int start, int end, int[][] g) {
if (start == end) {
a.add(start);
res.add(new ArrayList<>(a));
a.remove(a.size() - 1);
return;
}
for (int temp : g[start]) {
a.add(start);
dfs(res, a, temp, end, g);//搜索当前结点所指向的下一个结点
a.remove(a.size() - 1);
}
}
}