leetcode 124. Binary Tree Maximum Path Sum

Binary Tree Maximum Path Sum

题目描述

找出尽可能使得和最大的路径（所有点需要连接）

解题思路

设计递归函数findPathSum(node)，传入节点，节点为空时停止递归，返回到传入节点（左右子节点）为止积累的最大和。

如图左半部分，每次计算三个值：
sumLeft：根节点和左子树的和
sumRight：根节点和右子树的和
sumAll：根节点和左右子树的和
计算当前可能的最大和：
curMax = max(sumLeft, sumRight, sumAll, root.val)
maxSoFar = max(curMax, maxSoFar)

在图右半部分，递归调用findPathSum()只能返回max(sumLeft, sumRight, root.val)，不应包括蓝色的线路。

代码

class Solution:
    def __init__(self):
        self.maxSoFar = float("-inf") #全局变量
    
    def maxPathSum(self, root: TreeNode) -> int:
        self.findPathSum(root)
        return self.maxSoFar
        
    def findPathSum(self, root):
        if(root == None):
            return 0
        left = self.findPathSum(root.left) # maxSum till left node
        right = self.findPathSum(root.right) # maxSum till right node
        sumLeft = left + root.val
        sumRight = right + root.val
        sumAll = left + right + root.val 
        
        curMax = max(root.val, max(sumAll, max(sumLeft, sumRight)))

        self.maxSoFar = max(curMax, self.maxSoFar)
            
        return max(max(sumLeft, sumRight), root.val)