美文网首页
797. 所有可能的路径(Python)

797. 所有可能的路径(Python)

作者: 玖月晴 | 来源:发表于2021-01-27 11:17 被阅读0次

    难度:★★★☆☆
    类型:图
    方法:深度优先搜索

    力扣链接请移步本题传送门
    更多力扣中等题的解决方案请移步力扣中等题目录

    题目

    给一个有 n 个结点的有向无环图,找到所有从 0 到 n-1 的路径并输出(不要求按顺序)

    二维数组的第 i 个数组中的单元都表示有向图中 i 号结点所能到达的下一些结点(译者注:有向图是有方向的,即规定了 a→b 你就不能从 b→a )空就是没有下一个结点了。

    示例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]]

    示例 3:

    输入:graph = [[1],[]]
    输出:[[0,1]]

    示例 4:

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

    示例 5:

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

    提示:

    结点的数量会在范围 [2, 15] 内。
    你可以把路径以任意顺序输出,但在路径内的结点的顺序必须保证。

    解答

    这个题是图遍历的经典题,可以使用深度优先搜索方法实现。

    定义深度优先搜索函数get_paths_from(),函数的输入是起始节点的编号node,函数输出是以node节点为起点,到达终止节点dst_node的所有可能的路径,以嵌套列表的形式返回。函数完成的功能即为路径搜寻。

    一般使用深度优先搜索函数会在入口处添加终止条件作为递归终点,这道题的终止条件即为到达终点。也就是说,当起点结点node即为终点结点dst_node时,说明成功到达终点,返回这条唯一的路径[[dst_node]],注意这里要嵌套两层列表,因为内部的列表表示一条路径。

    当输入节点node不满足递归终点时(也是大多数情况),我们就要搜寻node节点的下一跳结点,可能有多个,然后遍历这些下一跳结点next_node,以next_node为起点进行递归调用,获得以next_node为起点,到达dst_node的所有路线,然后将这些路线加到当前结点node上。

    这里不用担心如果不存在node到dst_node的路线时的情况。因为如果不存在这些情况,路线列表是空的,不会进行遍历。

    其他变量说明:
    src_node,dst_node:起点和终点结点编号。
    nums:图中结点总数。

    
    class Solution(object):
        def allPathsSourceTarget(self, graph):
            nums = len(graph)
            src_node, dst_node = 0, nums - 1
    
            def get_paths_from(node):
                if node == dst_node:
                    return [[dst_node]]
                ans = []
                for next_node in graph[node]:
                    for path in get_paths_from(next_node):
                        ans.append([node] + path)
                return ans
    
            return get_paths_from(src_node)
    

    如有疑问或建议,欢迎评论区留言~

    有关更多力扣中等题的python解决方案,请移步力扣中等题解析

    相关文章

      网友评论

          本文标题:797. 所有可能的路径(Python)

          本文链接:https://www.haomeiwen.com/subject/xkgqzktx.html