【回溯】【leetcode】所有可能的路径

题目:

给一个有 n 个结点的有向无环图,找到所有从 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

来源:

797. 所有可能的路径

解题思路:回溯

注意:有向无环图graph[i]表示节点i能够到达的节点集合。

定义一个数组path,存放节点0开始能够到达的所有节点,并且先将节点0放入数组。

  • 递归终止条件:当path最后一个数字是最大值时
  • 结果满足条件:递归终止时
class Solution {
public:
    vector< vector<int> > result;
    vector<int> path;
    vector< vector<int> > allPathsSourceTarget(vector<vector<int>>& graph) {
        path.push_back(0);
        back(graph);
        return result;
    }
    void back(const vector< vector<int> >& graph) {
        int pre = path[path.size() - 1];
        if (pre == graph.size() - 1) {
            result.push_back(path);
            return;
        }
        // 上一节点能够到达的节点集合,每个都尝试一次
        for (int i = 0; i < graph[pre].size(); i++) {
            path.push_back(graph[pre][i]);
            back(graph);
            path.pop_back();
        }
    }
};

猜你喜欢

转载自blog.csdn.net/hbuxiaoshe/article/details/115046629