平衡二叉树的条件:为空或者子树高度之差的绝对值小于等于1,且子树也是平衡二叉树
最直接的做法,遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
if(root == null ) return true;
int h = height(root.left) - height(root.right);
return Math.abs(h) == 1 || h == 0;
}
//获取以node为根节点的树的高度
int height(TreeNode node){
if(node == null) return 0;
return Math.max(height(node.left), height(node.right)) + 1;
}
}
但上述的做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
public class Solution {
public boolean IsBalanced_Solution(TreeNode root) {
return height(root) != -1;
}
int height(TreeNode root){
if(root == null) return 0;
int left = height(root.left);
if(left == -1) return -1;
int right = height(root.right);
if(right == -1) return -1;
return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
}
}