要求
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1 。
示例 1:
输入:root = [3,9,20,null,null,15,7]
输出:true
示例 2:
输入:root = [1,2,2,3,3,null,null,4,4]
输出:false
示例 3:
输入:root = []
输出:true
核心代码
class TreeNode:
def __init__(self, val=0, left=None, right=None):
self.val = val
self.left = left
self.right = right
class Solution:
def isBalanced(self, root: TreeNode) -> bool:
if not root:
return True
def getHeight(node,h):
if not node:
return h
h += 1
return max(getHeight(node.left,h),getHeight(node.right,h))
l_height = getHeight(root.left,0)
r_height = getHeight(root.right,0)
return abs(l_height - r_height) <= 1 and self.isBalanced(root.left) and self.isBalanced(root.right)
解题思路:第一种解法:对于每个node,都判断一下其左右子树的相对高度,然后根据相对高度判断这个node满不满足条件,再递归地判断其左右孩子满不满足条件。这种思路会重复计算高度,所以会慢。