1、题目描述
https://leetcode-cn.com/problems/binary-tree-maximum-path-sum/
给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。
2、代码详解
# Definition for a binary tree node.
class TreeNode(object):
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution(object):
ret = float('-inf')
def maxPathSum(self, root):
"""
:type root: TreeNode
:rtype: int
"""
self.res = float("-inf") # 最妙的地方就是在递归中利用全局变量,来更新最大路径的值
# 后序遍历树,返回经过root的单边最大路径和,并维护整棵树的最大路径和
def helper(root):
# 终止条件
if not root:
return 0
# 若左右分支返回的值为负数,则对路径和做负贡献,直接舍弃, >0 时说明才能使路径变大
# 计算左边分支最大值,左边分支如果为负数还不如不选择
left = max(0, helper(root.left))
# 计算右边分支最大值,右边分支如果为负数还不如不选择
right = max(0, helper(root.right))
# 和全局变量比较,left->root->right 作为路径与历史最大值做比较
self.res = max(self.res, left + root.val + right)
return root.val + max(left, right) # 只返回包含当前根节点和左子树或者右子树的路径,
# 返回经过root的单边最大分支给上游(左子树和右子树产生的值较大的一个)
helper(root)
return self.res