563. 二叉树的坡度

给定一个二叉树,计算整个树的坡度。

一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。

整个树的坡度就是其所有节点的坡度之和。

示例:

输入: 
         1
       /   \
      2     3
输出: 1
解释: 
结点的坡度 2 : 0
结点的坡度 3 : 0
结点的坡度 1 : |2-3| = 1
树的坡度 : 0 + 0 + 1 = 1

注意:

  1. 任何子树的结点的和不会超过32位整数的范围。
  2. 坡度的值不会超过32位整数的范围。
# Definition for a binary tree node.
class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

class Solution:
    # 由于叶子结点的坡度为0,根结点的坡度为左右结点各自的和之差,所以符合后序遍历的特点
    # 注意这里采用列表res[0],可变对象作为“传址调用”或者用"self.res"
    def findTilt(self, root: TreeNode) -> int:
        res = [0]
        self.post_order_trans(root, res)
        return res[0]

    def post_order_trans(self, node: 'TreeNode', res: 'List[int]')->'int':
        if node is None:return 0
        left_sum = self.post_order_trans(node.left, res)
        right_sum = self.post_order_trans(node.right, res)
        res[0] += abs(left_sum - right_sum)
        print('(',left_sum, right_sum,left_sum + right_sum + node.val, res[0], ')', end= ' ')
        return left_sum + right_sum + node.val


if __name__ == '__main__':
    s = Solution()
    root = TreeNode(5)
    root.left = TreeNode(4)
    root.right = TreeNode(8)
    root.left.left = TreeNode(11)
    root.left.left.left = TreeNode(7)
    root.left.left.right = TreeNode(2)
    root.right.left = TreeNode(13)
    root.right.right = TreeNode(4)
    root.right.right.right = TreeNode(1)
    '''
    示例: 
    给定如下二叉树

                  5
                 / \
                4   8
               /   / \
              11  13  4
             /  \      \
            7    2      1
    
    '''
    print(s.findTilt(root))

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值