如何求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二叉树的所有路径和的方法了吗?如果还想学到更多技能或想了解更多相关内容,欢迎关注亿速云行业资讯频道,感谢各位的阅读!
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。
原文链接:https://my.oschina.net/tplinuxhyh/blog/4545647