LeetCode刷题之路：110. 平衡二叉树

如果帮助到您，还请点个关注吧，hahaha

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

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

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：true
示例 2：

输入：root = [1,2,2,3,3,null,null,4,4]
输出：false
示例 3：

输入：root = []
输出：true

#最直观的思路
第一种方法是从顶至底的方法，这种方法的弊端是存在许多重复运算：
构造一个获取当前节点最大深度的方法 height(root)，
通过比较此子树的左右子树的最大高度差abs(height(root.left) - height(root.right))，
来判断此子树是否是二叉平衡树。
注意：若树的所有子树都平衡时，此树才平衡。所以还需要判断当前节点的子树是否平衡
同时，若树跟节点为空的话，则直接返回True
若不为空则就要判断当前子树是否平衡 and 当前子树的左子树是否平衡 and 当前子树的右子树是否平衡

height方法：用于计算树的最大高度
终止条件：当node为空时，则返回高度0，说明当前树已经遍历完毕
返回值：返回左右子树的最大高度+1

第二种方法是从底向上遍历，若判断出某子树不是平衡树，则直接返回-1，剩下的节点就不需要判断了
假设当前节点 node 的左右子树高度差<2,
则返回以node为根节点的子树的最大高度，即max(self.height(node.left), self.height(node.right)) + 1
若当前节点node 左右子树高度差 >= 2,则返回-1，代表此子树不是平衡树
最后主函数里判断返回值是否为-1，若为-1 则返回 Fals，反之则为True

class Solution:
    def isBalanced(self, root: TreeNode) -> bool:
        '''
        if not root:
            return True
        return abs(self.height(root.left) - self.height(root.right)) < 2 and self.isBalanced(root.left) and self.isBalanced(root.right)   
    def height(self, node):
        if not node:
            return 0
        return max(self.height(node.left), self.height(node.right)) + 1
        '''
        return self.ban(root) != -1
    def ban(self, root):
        if not root:
            return 0
        left = self.ban(root.left)
        if left == -1:
            return -1
        right = self.ban(root.right)
        if right == -1:
            return -1
        return max(left, right) + 1 if abs(left - right) < 2 else -1