给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过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)