二叉树 7.8-二叉平衡树判断

题目：

有一棵二叉树，请设计一个算法判断这棵二叉树是否为平衡二叉树。

给定二叉树的根结点root，请返回一个bool值，代表这棵树是否为平衡二叉树。

思路：

平衡二叉树要求左右子树高度最大相差为1。那么就有三种可能：1. 左子树或右子树不是平衡二叉树；2. 左右子树都是平衡二叉树，但高度相差大于1；3. 左右子树都是平衡二叉树且高度相差小于等于1。因此不仅需要知道左右子树是否是平衡二叉树，还需要知道他们的高度。我们可以用一个int值来判断：如果子树不是平衡二叉树，则返回-1，否则返回子树高度（或深度）。

代码：

# -*- coding:utf-8 -*-

# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class CheckBalance:
    def check(self, root):
        # write code here
        def isAVL(root, h):
            if not root:
                return h-1
            left_height = isAVL(root.left, h+1)
            if left_height < 0:
                return -1
            right_height = isAVL(root.right, h+1)
            if right_height < 0:
                return -1
            if abs(left_height-right_height) > 1:
                return -1
            return max(left_height, right_height)
        
        res = isAVL(root, 1)
        return res >= 0