leetcode-563. 二叉树的坡度

题目

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

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

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

示例:

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

注意:

任何子树的结点的和不会超过32位整数的范围。
坡度的值不会超过32位整数的范围。

解题思路

后序遍历，树形dp，每个结点保存左子树结点之和、右子树结点之和、本结点tilt

成绩

时间>48.18%
空间>6.35%

代码

# 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 findTilt(self, root):
        """
        :type root: TreeNode
        :rtype: int
        """
        if not root:
            return 0
        node_info = {}
        stack = [(0, root)]
        while stack:
            stat, p = stack.pop()
            if stat == 0:
                stack.append((1, p))
                if p.left:
                    stack.append((0, p.left))
                if p.right:
                    stack.append((0, p.right))
            else:
                if p.left and p.right:
                    left_sum = sum(node_info[p.left][:2]) + p.left.val
                    right_sum = sum(node_info[p.right][:2]) + p.right.val
                    tilt = abs(left_sum - right_sum)
                    node_info[p] = [left_sum, right_sum, tilt]
                elif p.left:
                    left_sum = sum(node_info[p.left][:2]) + p.left.val
                    right_sum = 0
                    tilt = abs(left_sum - right_sum)
                    node_info[p] = [left_sum, right_sum, tilt]
                elif p.right:
                    left_sum = 0
                    right_sum = sum(node_info[p.right][:2]) + p.right.val
                    tilt = abs(left_sum - right_sum)
                    node_info[p] = [left_sum, right_sum, tilt]
                else:
                    node_info[p] = [0, 0, 0]
        tree_tilt = 0
        for node, val_list in node_info.items():
            tree_tilt += val_list[2]
        return tree_tilt