Leetcode 797. 所有可能的路径 C++

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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。