二叉平衡树(Balanced Binary Tree),也称为 AVL 树,是一种自平衡的二叉搜索树。它的左右子树的高度差不能超过1。
判断一棵二叉树是否为二叉平衡树,可以采用递归的方式。
对于每个节点,先判断其左右子树是否为平衡树,如果是,则计算左右子树的高度差,如果高度差不超过1,则返回当前节点所在子树的高度;
如果高度差超过1,则返回一个标记值表示当前子树不是平衡树。
具体实现可以定义一个函数,用于判断当前子树是否为平衡树,并返回当前子树的高度,如果不是平衡树,则返回一个标记值-1。
代码示例:
class TreeNode {
int val;
TreeNode left;
TreeNode right;
TreeNode(int x) { val = x; }
}
class Solution {
public boolean isBalanced(TreeNode root) {
return getHeight(root) != -1;
}
private int getHeight(TreeNode node) {
if (node == null) {
return 0;
}
int leftHeight = getHeight(node.left);
if (leftHeight == -1) {
return -1;
}
int rightHeight = getHeight(node.right);
if (rightHeight == -1) {
return -1;
}
if (Math.abs(leftHeight - rightHeight) > 1) {
return -1;
}
return Math.max(leftHeight, rightHeight) + 1;
}
}
在该代码中,isBalanced() 函数调用 getHeight() 函数获取当前节点所在子树的高度,并判断当前子树是否为平衡树。
getHeight() 函数采用递归的方式计算当前子树的高度,如果左右子树高度差超过1,则返回-1表示当前子树不是平衡树,否则返回当前子树的高度。
如果左右子树都是平衡树,并且高度差不超过1,则返回当前子树的高度。