判断一棵二叉树是否是平衡二叉树

输入一棵节点数为 n 二叉树，判断该二叉树是否是平衡二叉树。
在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树
平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。
样例解释：

样例二叉树如图，为一颗平衡二叉树
注：我们约定空树是平衡二叉树。

数据范围：,树上节点的val值满足 
要求：空间复杂度，时间复杂度 

class TreeNode:
    def __init__(self, x):
        self.val = x
        self.left = None
        self.right = None

def isBalanced(root: TreeNode) -> bool:
    def check(node):
        if node is None:
            return 0, True
        
        left_height, left_balanced = check(node.left)
        right_height, right_balanced = check(node.right)
        
        # 当前节点的高度等于左右子树的最大高度加1
        current_height = max(left_height, right_height) + 1
        
        # 如果左右子树都是平衡的，并且高度差不超过1，则当前子树也是平衡的
        is_balanced = left_balanced and right_balanced and abs(left_height - right_height) <= 1
        
        return current_height, is_balanced
    
    # 只需判断根节点是否平衡即可
    _, result = check(root)
    return result