思路
对于任意一个节点, 如果最大和路径包含该节点,只有两种情况:
1)当前节点的值+左、右子树最大路径和中,较大的那个值,再向父节点回溯,构成最大路径
2)左右子树都在最大路径中,加上该节点的值,构成最终的最大路径
class Solution(object):
def maxPathSum(self, root):
"""
:type root: TreeNode
:rtype: int
"""
self.res = float('-inf')
self.helper(root)
return self.res
#递归思路:记录左子树和右子树中路径更大的那个,并向父节点提供当前节点和子树组成的最大值
def helper(self, root):
#递归终止条件
if not root:
return 0
#每一步要做什么:记录1)情况的最大值,2)当前节点最大值 = root.val + left + right
left = max(0, self.helper(root.left)) # 如果子树路径和为负则应当置0表示最大路径不包含子树
right = max(0, self.helper(root.right))
self.res = max(self.res, root.val + left + right)
#返回值:此节点与左右子树中较大值之和
return root.val + max(right, left)
另一种版本:
class Solution:
def maxPathSum(self, root: Optional[TreeNode]) -> int:
self.max = float('-inf')
new_max = self.helper(root)
return max(self.max, new_max) # 2种情况再比大小
def helper(self, root):
if not root:
return float("-inf")
leftSum = self.helper(root.left)
rightSum = self.helper(root.right)
self.max = max(self.max, root.val + leftSum + rightSum, leftSum, rightSum) # 不可累加的1,5,6
return max(root.val, root.val + leftSum, root.val + rightSum) # 可累加的2,3,4
543 二叉树的直径
思路和124题一致
class Solution:
def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
self.res = -float("inf")
def helper(root):
if not root:
return 0
left = helper(root.left)
right = helper(root.right)
self.res = max(self.res, left + right + 1) # 左+右+当前节点
return max(left, right) + 1 # max(左,右)+当前根节点,回溯
helper(root)
return self.res - 1 # 距离为节点数量-1