LeetCode: 110.平衡二叉树

上一道题是建立一个高度平衡二叉树，下一道题目就是判断一棵树是否是平衡二叉树。

平衡二叉树的定义是：

一个树的每一个节点作为根节点，其左右子树的高度相差小于等于1。树的高度是指树的最大深度。

 

惭愧，我也是看了题解才明白的。然后自己实现了暴露法，和题解写法不同。我们一一来看。

有两种方法。第一种方法就是自顶而下深度优先遍历，先求出节点的左子树和右子树的高度，然后判断高度差是否满足条件，如果满足，则说明以这个节点为根节点的树是高度平衡的，进而判断这个节点的左节点和右节点是否平衡的，如果不满足，返回False。然后使用递归遍历每一个节点。

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        if root == None:
            return True
        def depth(root, num):
            if root == None:
                return num-1
            else:
                return max(depth(root.left, num+1), depth(root.right, num+1))
        
        if abs(depth(root.right, 2) - depth(root.left, 2)) <= 1:
            return self.isBalanced(root.left) and self.isBalanced(root.right)
        else:
            return False

我的官方题解写法是不同的。我注意到官方写法，求树深度是这样写的

def depth(root):
    if root == None:
        return 0
    else:
        return max(depth(root.right), depth(root.left)) + 1

观察得知，我通过把高度写到函数参数列表中和题解使用返回值+1的形式，都能得到目的，学到了。

 

第二种更优的办法是，从底至顶深度优先遍历，也称为提前阻断法，如果底层的节点都不满足是平衡二叉树了，那整棵树都不会是了，就可以提前退出了，顾称提前阻断法。

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        if root == None:
            return True
        if self.depth(root) == -1:
            return False
        else:
            return True
    
    def depth(self, root):
        if not root:
            return 0
        left = self.depth(root.left)
        if left == -1:
            return -1
        right = self.depth(root.right)
        if right == -1:
            return -1
        if abs(left - right) >1:
            return -1
        else:
            return max(left, right) +1 

区别于第一种方法，就是在递归的时候，一直遍历到底层，加了一个判断条件，判断底层节点是否是满足平衡二叉树的定义。不满足则返回 -1，递归返回-1。