Leetcode 797. 所有可能的路径
题目
给一个有 n 个结点的有向无环图,找到所有从 0 到 n-1 的路径并输出(不要求按顺序)
二维数组的第 i 个数组中的单元都表示有向图中 i 号结点所能到达的下一些结点(译者注:有向图是有方向的,即规定了a→b你就不能从b→a)空就是没有下一个结点了。
示例:
输入: [[1,2], [3], [3], []]
输出: [[0,1,3],[0,2,3]]
解释: 图是这样的:
0--->1
| |
v v
2--->3
这有两条路: 0 -> 1 -> 3 和 0 -> 2 -> 3.
提示:
- 结点的数量会在范围 [2, 15] 内。
- 你可以把路径以任意顺序输出,但在路径内的结点的顺序必须保证。
题解
dfs深搜回溯,从节点0开始遍历,终点为n-1。详细过程见代码
代码
vector<vector<int>> ans;
vector<int> visit; //标记已访问节点
void dfs(vector<int>& rote,int target,vector<vector<int>>& graph){
if(rote.back() == target){
ans.push_back(rote);
}else{
int now = rote.back();
for(int i=0; i<graph[now].size(); i++){
if(!visit[graph[now][i]]){
visit[graph[now][i]] = 1;
rote.push_back(graph[now][i]);
dfs(rote,target,graph);
rote.pop_back();
visit[graph[now][i]] = 0;
}
}
}
}
vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
int n = graph.size();
visit = vector<int>(n,0);
vector<int> now;
now.push_back(0);
visit[0] = 1;
dfs(now,n-1,graph);
return ans;
}
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/all-paths-from-source-to-target
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。