【LeetCode-797】所有可能的路径

问题

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

二维数组的第 i 个数组中的单元都表示有向图中 i 号结点所能到达的下一些结点(译者注:有向图是有方向的,即规定了 a→b 你就不能从 b→a )空就是没有下一个结点了。

示例

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

解答

class Solution {
public:
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        ans.push_back(0);
        dfs(graph, 0);
        return res;
    }
private:
    vector<vector<int>> res;
    vector<int> ans;
    void dfs(vector<vector<int>>& graph, int pos) {
        if (pos == graph.size() - 1) res.push_back(ans);
        for (int i : graph[pos]) {
            ans.push_back(i);
            dfs(graph, i);
            ans.pop_back();
        }
    }
};
posted @ 2021-08-01 13:33  tmpUser  阅读(25)  评论(0编辑  收藏  举报