【LeetCode】110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

    3
   / \
  9  20
    /  \
   15   7

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

       1
      / \
     2   2
    / \
   3   3
  / \
 4   4

返回 false 。

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time: 2018/7/31
# @Author: xfLi
# The file...
class TreeNode(object):
    def __init__(self, val=None, left=None, right=None):
        self.val = val
        self.lchild = left
        self.rchild = right

def isBalanced(root):
    """
    :type root: TreeNode
    :rtype: bool
    """
    if root is None:
        return True
    left_deep = getDeepth(root.lchild)
    right_deep = getDeepth(root.rchild)
    if abs(right_deep - left_deep) > 1:
        return False
    return isBalanced(root.lchild) and isBalanced(root.rchild)

def getDeepth(root): # 获取二叉树深度
    if root is None:
        return 0
    left_deep = getDeepth(root.lchild)
    right_deep = getDeepth(root.rchild)
    return max(left_deep, right_deep) + 1

if __name__ == '__main__':
    root = TreeNode(val=1, left=TreeNode(2, left=TreeNode(3), right=TreeNode(4)),
                    right=TreeNode(2, left=TreeNode(3), right=TreeNode(4)))
    result = isBalanced(root)
    print(result)