给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
class Solution:
def maxPathSum(self, root):
"""
:type root: TreeNode
:rtype: int
"""
def max_gain(node):
nonlocal max_sum
if not node:
return 0
# 左右字数的最大和
left_gain = max(max_gain(node.left), 0)
right_gain = max(max_gain(node.right), 0)
# 该节点的最大和
price_newpath = node.val + left_gain + right_gain
# 更新最大节点
max_sum = max(max_sum, price_newpath)
# 返回最大和
return node.val + max(left_gain, right_gain)
max_sum = float('-inf')
max_gain(root)
return max_sum