题目:
给定一个有 n 个节点的有向无环图,用二维数组 graph 表示,请找到所有从 0 到 n-1 的路径并输出(不要求按顺序)。
graph 的第 i 个数组中的单元都表示有向图中 i 号节点所能到达的下一些结点(译者注:有向图是有方向的,即规定了 a→b 你就不能从 b→a ),若为空,就是没有下一个节点了。
示例 :
输入:graph = [[1,2],[3],[3],[]]
输出:[[0,1,3],[0,2,3]]
解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3
思路:
深度优先,效率比较差
复杂度:
时间:O(n+m)。
空间:O(n)。
代码:
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
//题目是从0节点到n-1节点,用dfs更合适。然后用递归实现
List<List<Integer>> res = new LinkedList<>();
List<Integer> path = new LinkedList<>();
//调用dfs
dfs(0,graph,path,res);
return res;
}
private void dfs(int source,int[][] graph,List<Integer> path,List<List<Integer>> res){
//将source填入path
path.add(source);
//长度等了就加
if(source == graph.length-1){
res.add(new LinkedList<Integer>(path));
}else{
for(int next:graph[source]){
dfs(next,graph,path,res);
}
}
path.remove(path.size()-1);
}