题目:
给定一个二叉树,判断它是否是高度平衡的二叉树。
本题中,一棵高度平衡二叉树定义为:
一个二叉树的每个节点的左右两个子树的高度差的绝对值不超过 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
思路:
从根节点开始遍历,确保每一个节点的左右子树高度差值的绝对值不超过 1 。
涉及两次递归,一次为判断是否平衡求出当前左右子树的高度,另一次是当前节点平衡,接着求子节点是否为平衡二叉树。
代码实现:
// 通过 getHeight 方法求得每个子树的高度
public int getHeight (TreeNode root) {
if (root == null) {
return 0;
}
int leftHeight = getHeight(root.left);
int rightHeight = getHeight(root.right);
// 当前节点为根节点的树的高度,根节点加左右节点中较高的那一个
return 1 + (leftHeight > rightHeight ? leftHeight
: rightHeight);
}
public boolean isBalanced (TreeNode root) {
// 空树为平衡二叉树
if (root == null) {
return true;
}
// 只有一个根节点也是平衡二叉树
if (root.left == null && root.right == null) {
return true;
}
int leftHeight = getHeight(root.left);
int rightHeight = getHeight(root.right);
// 如果左右子树高度差值的绝对值 “大于1” 则返回 false
if (leftHeight - rightHeight > 1 ||
leftHeight - rightHeight < -1) {
return false;
}
// 递归判断左右子树是否平衡
return isBalanced(root.left) && isBalanced(root.right);
}