leetcode 1026

此题是想找到根节点和子节点的最大差值。关键是要找到子结构。可以发现最大差值是由每个子树的最大值和最小值决定的。因此我们新建一个函数,返回最大差值以及最大值最小值。当返回更上一层时,我们就可以更新这些值,并且根据最大最小值重新计算最大差值。

代码如下:

# 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):
    def tempDiff(self, root):
        if root.left == None and root.right == None:
            return 0, root.val, root.val
        if root.right == None:
            left_diff, left_max, left_min = self.tempDiff(root.left)
            new_diff = max(abs(root.val -left_max), abs(root.val- left_min))
            if new_diff < left_diff:
                new_diff = left_diff
            new_max = root.val
            new_min = root.val
            if new_max < left_max:
                new_max = left_max
            if new_min > left_min:
                new_min = left_min
            return new_diff, new_max, new_min
        if root.left == None:
            right_diff, right_max, right_min = self.tempDiff(root.right)
            new_diff = max(abs(root.val -right_max), abs(root.val- right_min))
            if new_diff < right_diff:
                new_diff = right_diff
            new_max = root.val
            new_min = root.val
            if new_max < right_max:
                new_max = right_max
            if new_min > right_min:
                new_min = right_min
            return new_diff, new_max, new_min
        more_diff, more_max, more_min = self.tempDiff(root.left)
        right_diff, right_max, right_min = self.tempDiff(root.right)
        if right_diff > more_diff:
            more_diff = right_diff
        if right_max > more_max:
            more_max = right_max
        if more_min > right_min:
            more_min = right_min
        new_diff = max(abs(root.val -more_max), abs(root.val- more_min))
        if new_diff < more_diff:
            new_diff = more_diff
        new_max = root.val
        new_min = root.val
        if new_max < more_max:
            new_max = more_max
        if new_min > more_min:
            new_min = more_min
        return new_diff, new_max, new_min
    def maxAncestorDiff(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if root == None:
            return 0
        result, _,_ = self.tempDiff(root)
        return result
        
        

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值