如何求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二叉树的所有路径和的方法了吗？如果还想学到更多技能或想了解更多相关内容，欢迎关注亿速云行业资讯频道，感谢各位的阅读！