每天一道leetcode：797. 所有可能的路径（图论&中等&深度优先遍历）

今日份题目：

给你一个有 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

示例2

输入：graph = [[4,3,1],[3,2,4],[3],[4],[]]
输出：[[0,4],[0,3,4],[0,1,3,4],[0,1,2,3,4],[0,1,4]]

提示

• n == graph.length

• 2 <= n <= 15

• 0 <= graph[i][j] < n

• graph[i][j] != i（即不存在自环）

• graph[i] 中的所有元素 互不相同

• 保证输入为 有向无环图（DAG）

题目思路

使用深度优先遍历，用p数组记录路径。递归遍历结束条件就是到达结尾，所以需要一个int数据记录当前所在位置，如果到结尾了就返回。

代码

class Solution 
{
public:
    vector> ans;
    vector p;

    void dfs(vector>& graph, int x, int n) 
    { //x用来标记当前所在位置，n标记结尾所在位置
        if(x==n) //到结尾了，返回
        {
            ans.push_back(p);
            return;
        }
        for(auto& y:graph[x]) //遍历临界节点
        {
            p.push_back(y);
            dfs(graph,y,n);
            p.pop_back();//还原队列，确保其他dfs操作的正确进行
        }
    }

    vector> allPathsSourceTarget(vector>& graph) 
    {
        p.push_back(0);
        dfs(graph,0,graph.size()-1);
        return ans;
    }
};

提交结果

 欢迎大家在评论区讨论，如有不懂的代码部分，欢迎在评论区留言！