LeetCode 797. 所有可能的路径

Posted 2022-12-03 Blocking The Sky

给你一个有 n 个节点的 有向无环图（DAG），请你找出所有从节点 0 到节点 n-1 的路径并输出（不要求按特定顺序）

二维数组的第 i 个数组中的单元都表示有向图中 i 号节点所能到达的下一些节点，空就是没有下一个结点了。

译者注：有向图是有方向的，即规定了 a→b 你就不能从 b→a 。

class Solution 
public:
    bool visited[20];
    void init()
        for(int i=0;i<20;i++)
            visited[i]=false;
    
    int firstarc(vector<vector<int>>& graph,int v)//返回v相邻的第一个结点
        int p=-1;
        if(graph[v].size()>0)//如果与v相邻有结点则返回第一个结点，没有返回-1
            p=graph[v][0];
        return p;
    
    int nextarc(vector<vector<int>>& graph,int v,int p)
        for(int i=0;i<graph[v].size();i++)//如果v结点相邻的结点p不是最后一个,则返回下一个结点
            if(graph[v][i]==p&&i<graph[v].size()-1)
                return graph[v][i+1];
        
        return -1; //如果v结点相邻的结点p是最后一个,则返回-1;
    
    vector<vector<int>> result;
    void find_path(vector<vector<int>>& graph,int start,int end,vector<int>& path)
        path.push_back(start);
        visited[start]=true;
        if(start==end)//找到一条路径则输出
            result.push_back(path);
        
        int p;
        p=firstarc(graph, start);//找到start第一个相邻结点
        while(p>=0)
            if(visited[p]==false)//如果顶点p未访问，递归访问它
                find_path(graph,p,end,path);
            
            p=nextarc(graph,start,p);//p指向start的下一个相邻结点
        
        path.pop_back();
        visited[start]=false;//恢复环境，使该节点可重新使用
    
    vector<vector<int>> allPathsSourceTarget(vector<vector<int>>& graph) 
        vector<int> path;
        init();
        find_path(graph,0,graph.size()-1,path);//dfs搜索所有路径
        return result;
    
;