[LeetCode]563. Binary Tree Tilt解法及python代码

原题链接：https://leetcode.com/problems/binary-tree-tilt/

先来了解一下Tilt的描述，原题的描述是The tilt of a tree node is defined as the absolute difference between the sum of all left subtree node values and the sum of all right subtree node values. Null node has tilt 0.

翻译过来就是一个树节点的Tilt是该节点左子树的总和与右子树的总和的绝对差值。空节点上午Tilt是0。我的理解就是一个节点左右子树值得差异。树的题大部分都可以用递归的方法来解。递归的访问左右子树，然后返回节点值的总和用来计算每个节点的Tilt。最开始的想法是为了使解题过程清晰，就用一个list将所有计算出来的Tilt都记录下来，最后再计算和返回（其实是可以在递归的过程中直接算出最后结果的）。下面代码：

class Solution(object):
def findTilt(self, root):
"""
:type root: TreeNode
:rtype: int
"""
Tilt = []
def helper(node):
if not node.left and not node.right:
Tilt.append(0)
return node.val
if not node.left and node.right:
temp0 = helper(node.right)
Tilt.append(abs(temp0))
return node.val + temp0
if node.left and not node.right:
temp1 = helper(node.left)
Tilt.append(abs(temp1))
return node.val + temp1
if node.left and node.right:
temp2 = helper(node.left)
temp3 = helper(node.right)
Tilt.append(abs(temp2-temp3))
return node.val + temp2 + temp3
sum = 0
if not root:
return sum
helper(root)
print Tilt
for key in Tilt:
sum += key
return sum

上面这种方法会慢一些，但是整个的过程会很清晰，方便理解。

下面这种方法就是在递归的过程中返回一个二元组，(sum,TiltSum),递归的出口是在访问到空节点时返回(0,0)二元组。这种写法相对的快一些（我认为时间复杂度都是O(n),但实际的执行时间会差一些，代码也相对简洁）。下面是代码:

class Solution1(object):
def findTilt(self, root):
"""
:type root: TreeNode
:rtype: int
"""
#return (sum,TiltSum)
def helper(node):
if not node:
return (0,0)
left = helper(node.left)
right = helper(node.right)
return (node.val + left[0] + right[0],abs(left[0] - right[0]) + left[1] + right[1])
return helper(root)[1]