回溯题目：所有可能的路径

题目

标题和出处

标题：所有可能的路径

出处：797. 所有可能的路径

难度

4 级

题目描述

要求

给定一个有 $\text{n}$ 个结点的有向无环图，结点从 $\text{0}$ 到 $\text{n}-\text{1}$ 编号，找出所有从结点 $\text{0}$ 到结点 $\text{n}-\text{1}$ 的路径，按任意顺序返回答案。

图的表示如下：$\text{graph[i]}$ 是从结点 $\text{i}$ 可以访问的所有结点的列表（即从结点 $\text{i}$ 到结点 $\text{graph[i][j]}$ 存在一条有向边）。

示例

示例 1：

输入：$\text{graph = [[1,2],[3],[3],[]]}$
输出：$\text{[[0,1,3],[0,2,3]]}$
解释：有两条路径：$\text{0}\to \text{1}\to \text{3}$ 和 $\text{0}\to \text{2}\to \text{3}$。

示例 2：

输入：$\text{graph = [[4,3,1],[3,2,4],[3],[4],[]]}$
输出：$\text{[[0,4],[0,3,4],[0,1,3,4],[0,1,2,3,4],[0,1,4]]}$

数据范围

• $\text{n}=\text{graph.length}$
• $\text{2}\le \text{n}\le \text{15}$
• $\text{0}\le \text{graph[i][j]}<\text{n}$
• $\text{graph[i][j]}\ne \text{i}$（即不存在自环）
• $\text{graph[i]}$ 中的所有元素各不相同
• 保证输入为有向无环图

解法

思路和算法

这道题要求找到所有从结点 $0$ 到结点 $n-1$ 的路径。由于给定的图是有向无环图，因此图中的任意一条路径都不会访问同一个结点两次，路径的长度一定是有限的。可以使用回溯得到所有路径。

从结点 $0$ 出发。对于每个访问到的结点 $\text{node}$，执行如下操作。

1. 将 $\text{node}$ 添加到路径的末尾。

2. 判断 $\text{node}$ 是否等于 $n-1$，执行相应的操作。

   • 如果 $\text{node}=n-1$，则找到一条从结点 $0$ 到结点 $n-1$ 的路径，将路径添加到答案中。

   • 如果 $\text{node}\ne n-1$，则对于 $\text{graph}[\text{node}]$ 中的每个结点 $\text{next}$，递归地访问 $\text{next}$。

3. 将 $\text{node}$ 从路径的末尾移除。

遍历结束时，即可得到所有从结点 $0$ 到结点 $n-1$ 的路径。

代码

class Solution {
    List<List<Integer>> allPaths = new ArrayList<List<Integer>>();
    List<Integer> temp = new ArrayList<Integer>();
    int n;
    int[][] graph;

    public List<List<Integer>> allPathsSourceTarget(int[][] graph) {
        this.n = graph.length;
        this.graph = graph;
        backtrack(0);
        return allPaths;
    }

    public void backtrack(int node) {
        temp.add(node);
        if (node == n - 1) {
            allPaths.add(new ArrayList<Integer>(temp));
        } else {
            int[] nextNodes = graph[node];
            for (int next : nextNodes) {
                backtrack(next);
            }
        }
        temp.remove(temp.size() - 1);
    }
}

复杂度分析

• 时间复杂度：$O(n\times 2^n)$，其中 $n$ 是图中的结点数。路径数最多是 $2^n$，对于每条路径需要 $O(n)$ 的时间添加到答案中，因此时间复杂度是 $O(n\times 2^n)$。

• 空间复杂度：$O(n)$，其中 $n$ 是图中的结点数。存储当前路径的列表和递归调用栈需要 $O(n)$ 的空间。注意返回值不计入空间复杂度。