leetcode124. 二叉树中的最大路径和、543.二叉树的直径

思路

对于任意一个节点, 如果最大和路径包含该节点，只有两种情况：
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