797. 所有可能的路径 - 力扣(LeetCode)
给你一个有 n
个节点的 有向无环图(DAG),请你找出所有从节点 0
到节点 n-1
的路径并输出(不要求按特定顺序)
graph[i]
是一个从节点 i
可以访问的所有节点的列表(即从节点 i
到节点 graph[i][j]
存在一条有向边)。
示例 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]]
思路:
使用dfs深度优先算法,从起点0开始遍历当前可去节点有哪些,直到走到终点target
递归函数参数
当前节点cur,终点target,图信息graph,当前path经过节点记录,当前路径curPath,最后结果ret
void dfs(int cur,int target,vector<vector<int>>& graph,unordered_set<int>& setPath,vector<int>& curPath,vector<vector<int>>& ret)
递归终止条件:cur==target
递归内容:
循环进入当前节点可去的节点,寻找可行方案
for(int e:graph[cur]){if(setPath.count(e)) continue;setPath.insert(e);curPath.push_back(e);dfs(e,target,graph,setPath,curPath,ret);setPath.erase(e);curPath.pop_back();}
class Solution {
public:void dfs(int cur,int target,vector<vector<int>>& graph,unordered_set<int>& setPath,vector<int>& curPath,vector<vector<int>>& ret){if(cur==target){ret.push_back(curPath);return;}for(int e:graph[cur]){if(setPath.count(e)) continue;setPath.insert(e);curPath.push_back(e);dfs(e,target,graph,setPath,curPath,ret);setPath.erase(e);curPath.pop_back();}return;}vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {int target=graph.size()-1;//终点unordered_set<int> setPath;vector<int> curPath;//加入起点curPath.push_back(0);setPath.insert(0);vector<vector<int>> ret;dfs(0,target,graph,setPath,curPath,ret);return ret;}
};