剑指offer-二叉树

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def IsBalanced_Solution(self, pRoot):
        # write code here
        if not pRoot:
            return True
        res = abs(self.getDepth(pRoot.left) - self.getDepth(pRoot.right))
        if res <= 1 and self.IsBalanced_Solution(pRoot.left) and self.IsBalanced_Solution(pRoot.right):
            return True
        return False
        
    def getDepth(self, root):
        if not root:
            return 0
        if not (root.left or root.right):
            return 1
        return max(self.getDepth(root.left), self.getDepth(root.right)) + 1