如何求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二叉樹的所有路徑和的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！