leet-code/src/main/java/leetcode/editor/cn/AllPathsFromSourceToTarget.java

106 lines
2.8 KiB
Java
Raw Normal View History

2021-08-23 14:05:33 +08:00
//给你一个有 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
//
// Related Topics 深度优先搜索 广度优先搜索 图 回溯 👍 143 👎 0
package leetcode.editor.cn;
2021-08-25 11:54:31 +08:00
import javafx.util.Pair;
import java.util.*;
2021-08-23 14:05:33 +08:00
//797:所有可能的路径
class AllPathsFromSourceToTarget {
public static void main(String[] args) {
//测试代码
Solution solution = new AllPathsFromSourceToTarget().new Solution();
}
//力扣代码
//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
2021-08-25 11:54:31 +08:00
List<List<Integer>> result = new ArrayList<>();
Queue<Pair<Integer, List<Integer>>> queue = new LinkedList<>();
queue.add(new Pair<>(0, new ArrayList<>(Collections.singletonList(0))));
while (!queue.isEmpty()) {
Pair<Integer, List<Integer>> pair = queue.poll();
int index = pair.getKey();
List<Integer> paths = pair.getValue();
for (int gr : graph[index]) {
if (paths.contains(gr)) {
continue;
}
paths.add(gr);
if (gr == graph.length - 1) {
result.add(new ArrayList<>(paths));
paths.remove((Integer) gr);
continue;
}
queue.add(new Pair<>(gr, new ArrayList<>(paths)));
paths.remove((Integer) gr);
2021-08-23 14:05:33 +08:00
}
}
2021-08-25 11:54:31 +08:00
return result;
2021-08-23 14:05:33 +08:00
}
}
//leetcode submit region end(Prohibit modification and deletion)
}