[Python手撕]二叉樹中的最大路徑和

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxPathSum(self, root: Optional[TreeNode]) -> int:

        def maxpath(root):
            # 葉子結點下面的空節點直接返回0
            if not root:
                return 0

            # 左邊的最大路徑和右邊的最大路徑
            # 如果有一邊的路徑是負數，那就丟掉這邊的路徑
            left = max(maxpath(root.left), 0)
            right = max(maxpath(root.right), 0)

            # 更新以當前節點為拐點的路徑
            # 如果所有節點都是負數，會這在裡保留最大的負數
            self.res = max(self.res, root.val + left + right)

            # 向上返回當前節點和較大的一個側邊
            return root.val + max(left, right)

        self.res = -float("inf")
        maxpath(root)
        return self.res