给你一个有 n 个节点的 有向无环图(DAG),请你找出所有从节点 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
示例 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]]
提示:
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/all-paths-from-source-to-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
用DFS来求所有可能的路径,从节点0出发,利用数组graph中的信息遍历所有能够到达的节点,每次经过一个节点就把该节点加入到路径数组path中,当抵达节点n-1时把当前路径加入ans数组。利用标记清除这一步骤,每次递归搜索完当前节点能够到达的节点后,把当前节点从path中删除。
时间复杂度\(O(n×2^n)\),空间复杂度O(n)。
class Solution { public: vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) { vector<vector<int>> ans; vector<int> path; path.push_back(0); dfs(graph, ans, path, 0, graph.size() - 1); return ans; } private: void dfs(vector<vector<int>>& graph, vector<vector<int>>& ans, vector<int>& path, int pos, int dst) { if(pos == dst) { ans.push_back(path); return; } for(const auto &x:graph[pos]) { path.push_back(x); dfs(graph, ans, path, x, dst); path.pop_back(); } } };