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

题目:

给定一个非空二叉树,返回其最大路径和。
本题中,路径被定义为一条从树中任意节点出发,沿父节点-子节点连接,达到任意节点的序列。该路径至少包含一个节点,且不一定经过根节点。

示例 1:

输入:[1,2,3]
在这里插入图片描述
输出:6

示例 2:

输入:[-10,9,20,null,null,15,7]
在这里插入图片描述
输出:42

思路:递归。

  参考官方题解

  为每一个节点定义一个最大贡献值,表示以该节点为根节点的子树中,经过该节点的最大路径和。根据这个定义,空节点的贡献为0,非空节点的贡献是节点值加上左右子节点中较大的贡献,举例如下,节点旁边的值是该节点的最大贡献值:
在这里插入图片描述

  这样,可以递归得到每一个节点的贡献值,同时,在递归的过程中,经过当前节点的最大路径和为,当前节点的值加上左右子节点的最大贡献(此处需要注意,如果子节点的贡献是负值,那我们选择不经过该子节点,也就是说不计入路径和)。维护一个最大路径和,在递归的过程中不断更新它。

代码:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def maxPathSum(self, root: TreeNode) -> int:
        maxSum = float("-inf")
        def getMaxSum(root):
            nonlocal maxSum
            if not root:
                return 0
            
            # 保证不取负值,例如[2, -1]
            maxLeft = max(getMaxSum(root.left), 0)
            maxRight = max(getMaxSum(root.right), 0)
            
            maxCur = root.val + max(maxLeft, maxRight)

            maxSum = max(maxSum, maxLeft + maxRight + root.val)
            
            return maxCur
        
        getMaxSum(root)
        return maxSum
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值