[LeetCode]797. 所有可能的路径

类似于字符串的全遍历

Posted by JinFei on August 25, 2021

题目描述

给你一个有 n 个节点的 有向无环图(DAG),请你找出所有从节点 0 到节点 n-1 的路径并输出(不要求按特定顺序)

二维数组的第 i 个数组中的单元都表示有向图中 i 号节点所能到达的下一些节点,空就是没有下一个结点了。

译者注:有向图是有方向的,即规定了 a→b 你就不能从 b→a 。

Example 1:

示例1 输入:graph = [[1,2],[3],[3],[]] 输出:[[0,1,3],[0,2,3]] 解释:有两条路径 0 -> 1 -> 3 和 0 -> 2 -> 3

Example 2:

示例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]]

Example 3:

输入:graph = [[1],[]] 输出:[[0,1]]

Constraints:

  • n == graph.length
  • 2 <= n <= 15
  • 0 <= graph[i][j] < n
  • graph[i][j] != i(即,不存在自环)
  • graph[i] 中的所有元素 互不相同
  • 保证输入为 有向无环图(DAG)

解题思路

  • DFS搜索,当路径到达一定个数时,就代表找到一个从原点到钟点的路径。
  • 类似于字符串的全排列

C++代码

class Solution {
public:
    // 这里很像字符串的全排列
    // 每个数组中取一个值,当取到指定个数时,就代表找到一条路径
    vector<int> path;
    vector<vector<int>> res;
    void funHelper(vector<vector<int>>& graph, int index, int size){
        if(index == size){
            res.push_back(path);
            return;
        }
        for(auto& i : graph[index]){
            path.push_back(i);
            funHelper(graph, i, size);
            path.pop_back();
        }
    }
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        path.push_back(0);
        funHelper(graph, 0, graph.size() - 1);
        return res;
    }
};