题解 | #判断是不是平衡二叉树#

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param pRoot TreeNode类 
# @return bool布尔型
# 自顶向下递归，自下而上返回
class Solution:
    def IsBalanced_Solution(self , pRoot: TreeNode) -> bool:
        # write code here
        return self.depth(pRoot) != -1 # 最终只需判断最上层返回的是不是-1即可。若是则不平衡，若不是则平衡，且返回值是树的高度

    def depth(self, root): # 递归出当前节点的高度
        if not root:
            return 0 # 直接到叶子节点，返回的是0
        left = self.depth(root.left) 
        if left == -1:
            return -1
        right = self.depth(root.right)
        if right == -1:
            return -1
        return max(left, right) + 1 if abs(left-right) < 2 else -1 # 返回左子高度和右子高度最高者+1，如果左右子不平衡了，直接返回-1，一直向上层返回-1