797. All Paths From Source to Target

仅供自己学习

 

思路：

这个题目给的graph是一个二维数组，例如，[[1,2],[3],[3],[]]，[1,2]是下标为0的位置的数组，所以是0结点的出度边的终点，就是0指向1，2.同样对于下标为1的位置是[3]，则1节点指向3.

那么我们就可以用DFS，从0结点开始从1深入，出来之后再从2深入即可，因为题目求的是0到n-1的路径，所以将一条路径加入到res二维数组的条件是当前结点cur=size()-1。

 

代码：

class Solution {
public:
    void DFS(vector<vector<int>>& graph,vector<vector<int>>&res,vector<int> path,int cur){
        path.push_back(cur);
        if(cur==graph.size()-1) res.push_back(path);
        else for(auto neighbor:graph[cur]) DFS(graph,res,path,neighbor);
    }
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        vector<vector<int>> res;
        DFS(graph,res,{},0);
        return res;
    }
    
};

 优化

class Solution {
public:
    vector<vector<int>> res;
    vector<int> path;
    void DFS(vector<vector<int>>& graph,int cur){
        path.push_back(cur);
        if(cur==graph.size()-1){
            res.push_back(path);
            return;
        }
        for(int i=0;i<graph[cur].size();++i){
            DFS(graph,graph[cur][i]);
            path.pop_back();
        }
    }
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) {
        DFS(graph,0);
        return res;
    }
    
};