千家信息网

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

发表于:2025-01-22 作者:千家信息网编辑
千家信息网最后更新 2025年01月22日,如何求python二叉树的所有路径和,相信很多没有经验的人对此束手无策,为此本文总结了问题出现的原因和解决方法,通过这篇文章希望你能解决这个问题。二叉树的所有路径和需求基于257变为求路径和给定一个二
千家信息网最后更新 2025年01月22日如何求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 = Noneclass 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二叉树的所有路径和的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注行业资讯频道,感谢各位的阅读!

0