给你一个有 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
class Solution {
public:
vector<int> path;
vector<vector<int>> result;
void dfs(int n, vector<vector<int>>& graph){
if (n == graph.size() - 1){
result.push_back(path);
return;
}
for (int i = 0; i < graph[n].size(); i++){
path.push_back(graph[n][i]);
dfs(graph[n][i], graph);
path.pop_back();
}
}
vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
path.push_back(0);
dfs(0, graph);
return result;
}
};
使用深搜三部曲来分析题目:
1.确认递归函数,参数
首先我们dfs函数一定要存一个图,用来遍历的,还要存一个目前我们遍历的节点,定义为n
至于 单一路径,和路径集合可以放在全局变量,那么代码是这样的:
vector<vector<int>> result; // 收集符合条件的路径
vector<int> path; // 0节点到终点的路径
// n:目前遍历的节点
// graph:存当前的图
void dfs (vector<vector<int>>& graph, int x)
2.确认终止条件
当目前遍历的节点 为 最后一个节点的时候,就找到了一条,从 出发点到终止点的路径。
当前遍历的节点,我们定义为n,最后一点节点,就是 graph.size() - 1
所以 但 xn等于 graph.size() - 1 的时候就找到一条有效路径。 代码如下:
// 要求从节点 0 到节点 n-1 的路径并输出,所以是 graph.size() - 1
if (n == graph.size() - 1) { // 找到符合条件的一条路径
result.push_back(path); // 收集有效路径
return;
}
3.处理目前搜索节点出发的路径
接下来是走 当前遍历节点x的下一个节点。
首先是要找到 n节点链接了哪些节点呢? 遍历方式是这样的:
for (int i = 0; i < graph[n].size(); i++) { // 遍历节点n链接的所有节点
接下来就是将 选中的n所连接的节点,加入到 单一路劲来。
path.push_back(graph[n][i]); // 遍历到的节点加入到路径中来
一些录友可以疑惑这里如果找到n 链接的节点的,例如如果n目前是节点0,那么目前的过程就是这样的:
二维数组中,graph[n][i] 都是n链接的节点,当前遍历的节点就是 graph[n][i]
。
进入下一层递归
dfs(graph, graph[n][i]); // 进入下一层递归
最后就是回溯的过程,撤销本次添加节点的操作。 该过程整体代码:
for (int i = 0; i < graph[n].size(); i++) { // 遍历节点n链接的所有节点
path.push_back(graph[n][i]); // 遍历到的节点加入到路径中来
dfs(graph, graph[n][i]); // 进入下一层递归
path.pop_back(); // 回溯,撤销本节点
}