[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