剑指 Offer II 110. 所有路径

剑指 Offer II 110. 所有路径

给定一个有 n 个节点的有向无环图，用二维数组 graph 表示，请找到所有从 0 到 n-1 的路径并输出（不要求按顺序）。
graph 的第 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
• 保证输入为有向无环图 (GAD)

​

注意：本题与主站 797 题相同：https://leetcode-cn.com/problems/all-paths-from-source-to-target/

思路

回溯法，用 map 存储 node 对应的 edge，采用回溯及深度优先遍历每一个路径，判断终止节点是否是 n-1 节点，是的话存入结果数组中，最后返回即可。

解答

/**
 * @param {number[][]} graph
 * @return {number[][]}
 */
const allPathsSourceTarget = (graph) => {
  const res = [],
    temp = [0];
  const graphMap = {};
  for (let i = 0; i < graph.length; i++) {
    graphMap[i] = new Set(graph[i]);
  }
  const visit = (i) => {
    const queue = [...graphMap[i]];
    if (i === graph.length - 1) {
      res.push(temp.slice());
    }
    while (queue.length) {
      const cur = queue.shift();
      temp.push(cur);
      visit(cur);
      temp.pop();
    }
  };
  visit(0);
  return res;
};