平衡二叉树:对于每一个节点,其左子树和右子树的高度差不超过1。
对于每一个节点,若其左子树和右子树非平衡,或者左子树和右子树的高度差超过1,则可以直接判断整棵树为非平衡
public boolean isBalanced(TreeNode root) {
return height(root) != -1;
}
public int height(TreeNode root){ //计算root的深度,如果root为非平衡,则返回-1,否则返回root的高度
if(root == null)
return 0;
int leftHeight = height(root.left);
int rightHeight = height(root.right);
if(leftHeight == -1 || rightHeight == -1 || Math.abs(leftHeight - rightHeight) > 1) //root为非平衡
return -1;
int height = leftHeight > rightHeight ? leftHeight + 1 : rightHeight + 1;
return height;
}