如何求python二叉树的所有路径和

如何求python二叉树的所有路径和,相信很多没有经验的人对此束手无策,为此本文总结了问题出现的原因和解决方法,通过这篇文章希望你能解决这个问题。

创新互联专注于古县网站建设服务及定制,我们拥有丰富的企业做网站经验。 热诚为您提供古县营销型网站建设,古县网站制作、古县网页设计、古县网站官网定制、重庆小程序开发服务,打造古县网络公司原创品牌,更为您提供古县网站排名全网营销落地服务。

二叉树的所有路径和

  • 需求基于257变为求路径和

给定一个二叉树,返回所有从根节点到叶子节点的路径。 说明: 叶子节点是指没有子节点的节点。

示例:

输入

1 /
2 3
5

输出: ["1->2->5", "1->3"]

解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3

代码思路

  • dfs

  • 叶子节点,获取本分支的路径, 求和

  • 汇总所有路径和

#!/usr/bin/env python
"""
doc:

"""
from typing import List, Set


# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None


class Solution:
    def dfs(self, root: TreeNode, path: List[str], path_map: List[str]):
        if not root:
            return None

        path.append(root.val)
        if root.left is None and root.right is None:
            # path_str = "->".join(path)
            path_sum = sum(path)
            print(f'--> cur leaf path:{path} sum:{path_sum}')
            path_map.append(path_sum)
            return

        self.dfs(root.left, list(path), path_map)
        self.dfs(root.right, list(path), path_map)

    def binaryTreePaths(self, root: TreeNode) -> List[str]:
        path = []
        path_map = []
        self.dfs(root, path, path_map)
        return sum(path_map)


def main():
    root = TreeNode(1)
    node_2 = TreeNode(2)
    root.left = node_2
    root.right = TreeNode(3)
    node_2.right = TreeNode(5)
    ret = Solution().binaryTreePaths(root)
    print(ret)


if __name__ == '__main__':
    main()

看完上述内容,你们掌握如何求python二叉树的所有路径和的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注创新互联行业资讯频道,感谢各位的阅读!


网站题目:如何求python二叉树的所有路径和
分享网址:http://azwzsj.com/article/ihpjio.html