LeetCode110. 平衡二叉树

1. 题目描述

题目来源：力扣

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 

2. 题解

2.1 题解思路

从根节点开始，判断一个节点的左右深度差是否超过1。分析：从根节点开始，分别获取左右子树的高度，那么根节点的左子节点也需判断其左右子树的高度差是否超过1，直到叶子节点位置。且根节点的左子树的高度为 max{左子树的左子树高度，左子树的右子树高度}。根据该分析，可知判断一棵树是否为平衡树的逻辑思路为：递归结构。

按照上述思路，返回值为一个节点左右子树的高度。我们想要得到一棵树是否为平衡树，那么上述过程可以作为一个子函数调用。难点在于，每个节点处判断左右子树高度差是否超过1。如果超过1，标记返回结果-1。如果当前节点的左子树的高度为-1，返回上一级函数时，仍返回-1；右子树同样。

按照此思路，算法如下：

内层函数：从根节点开始

（1）初始化当前节点的左右子度为Level_left = Level_right = 0

（2）递归调用左子节点的高度，若左子树的高度为-1，返回-1；

（3）递归调用右子节点的高度，若右子树的高度为-1，返回-1；

（4）若|左子树高度-右子树的高度| > 1，那么返回-1；

（5）返回该节点的深度为max{左子树的高度，右子树的高度}+1

外层函数：最后判断返回值如果为-1，那么不是平衡树，否则是平衡树。

2.2 代码

代码如下，

# Definition for a binary tree node.
class TreeNode(object):
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right

    def set_left(self, left):
        self.left = left

    def set_right(self, right):
        self.right = right

class Solution(object):
    def node_level(self, root):
        """
        :param node: TreeNode
        :return: node的左右深度
        """
        current = root
        L_left = 0
        L_right = 0
        if current.left != None:
            L_left = self.node_level(current.left)
        if L_left == -1:
            return -1

        if current.right != None:
            L_right = self.node_level(current.right)
        if L_right == -1:
            return -1
            
        if L_right - L_left > 1 or L_right - L_left < -1:
            return -1
        else:
            return max(L_left, L_right)+1

    def isBalanced(self, root):
        """
        :type root: TreeNode
        :rtype: bool
        """
        if root == None:
            return True
        if self.node_level(root) == -1:
            return False
        else:
            return True

if __name__ == '__main__':
    root = [1, 2, 2, 3, 4, 4, 3]
    root = [1,  2, 3]
    node1 = TreeNode(1)
    node2 = TreeNode(2)
    node3 = TreeNode(3)
    node4 = TreeNode(4)
    node5 = TreeNode(5)
    node6 = TreeNode(6)
    node7 = TreeNode(7)


    node1.set_left(node2)
    node1.set_right(node3)

    node2.set_left(node4)
    node4.set_left(node6)

    node3.set_right(node5)
    node5.set_right(node7)

    sol = Solution()
    r_level = sol.node_level(node1)
    r = sol.isBalanced(node1)
    print(r_level)
    print(r)

这其中有一个细节，l_left和l_right初始化为0还是1，然后在左节点非空的情况下，L_left是直接+1还是在判断完L_left==-1；应该是初始化为0，然后在判断子树及本身不是非平衡树的情况下，再—+1.