给定一个二叉树,计算整个树的坡度。
一个树的节点的坡度定义即为,该节点左子树的结点之和和右子树结点之和的差的绝对值。空结点的的坡度是0。
整个树的坡度就是其所有节点的坡度之和。
示例:
输入: 1 / \ 2 3 输出: 1 解释: 结点的坡度 2 : 0 结点的坡度 3 : 0 结点的坡度 1 : |2-3| = 1 树的坡度 : 0 + 0 + 1 = 1
注意:
- 任何子树的结点的和不会超过32位整数的范围。
- 坡度的值不会超过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))