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

# 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
       #如果节点为空，那么说明该节点不存在默认为True不需要进行比较
        if not pRoot:
            return True

#要比较左右字树的深度，那么必须有办法获取到左右字数的深度，获取深度的函数用GetDepth函数
#获取深度的函数的逻辑就是当根节点不为空时，深度为左子树的深度对比右边字数的深度哪个大取哪个，然后+1，当遍历到最后，为空时返回0，叶子节点就是空加1深度为1
        def GetDepth(root):

            if  root:
                return max(GetDepth(root.left),GetDepth(root.right))+1
            else:
                return 0
        #print(GetDepth(pRoot))
        
        if abs(GetDepth(pRoot.left)-GetDepth(pRoot.right))>1:
            return False
        else:
            return self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right)