【深度度优先搜索】leetcode 797. 所有可能的路径

最新推荐文章于 2024-07-10 22:35:57 发布

lele_ne

最新推荐文章于 2024-07-10 22:35:57 发布

阅读量200

点赞数

分类专栏： # 深度优先搜索算法之路文章标签： leetcode 深度优先算法

本文链接：https://blog.csdn.net/lele_ne/article/details/126563239

版权

算法之路同时被 2 个专栏收录

84 篇文章 0 订阅

订阅专栏

深度优先搜索

16 篇文章 0 订阅

订阅专栏

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 == g r a p h . l e n g t h$
$2 <= n <= 15$
$0 <= g r a p h [i] [j] < n$
$g r a p h [i] [j]! = i （即不存在自环）$
$g r a p h [i] 中的所有元素互不相同$
$保证输入为有向无环图（ D A G ）$

方法一：深度优先搜索

解题思路

我们可以使用深度优先搜索的方式求出所有可能的路径。具体地，我们从 0 号点出发，使用栈记录路径上的点。每次我们遍历到点 $n - 1$ ，就将栈中记录的路径加入到答案中。

特别地，因为本题中的图为有向无环图 $（\text{DAG}）$ ，搜索过程中不会反复遍历同一个点，因此我们无需判断当前点是否遍历过。

代码

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

复杂度分析

时间复杂度： $\times 2^n)$ 。其中 $n$ 为图中点的数量。我们可以找到一种最坏情况，即每一个点都可以去往编号比它大的点。此时路径数为 $O(2^n)$ ，且每条路径长度为 $O (n)$ ，因此总时间复杂度为 $\times 2^n)$ 。
空间复杂度： $O (n)$ 。其中 $n$ 为点的数量。主要为栈空间的开销。注意返回值不计入空间复杂度。