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

最新推荐文章于 2019-03-07 11:19:32 发布

jackhell0

最新推荐文章于 2019-03-07 11:19:32 发布

阅读量453

点赞数

分类专栏： leetcode 文章标签： python leetcode 递归

本文链接：https://blog.csdn.net/jackhell2/article/details/79238940

版权

leetcode 专栏收录该内容

7 篇文章

订阅专栏

本文解析了LeetCode上二叉树倾斜问题，通过两种不同的递归方法实现了解决方案，一种采用列表记录每个节点的倾斜值，另一种则在递归过程中直接计算总的倾斜值。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

原题链接：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]