假如有一个二叉树,需要确定它是否是平衡二叉树。最直接的做法,就是遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。
public boolean IsBalance_Solution(TreeNode root){
if(root==null) return true;
int left = depth(root.left);
int right = depth(root.right);
if(Math.abs(left-right)>1){
return false;
}
// return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
return true;
}
public int depth(TreeNode root){
if(root==null){
return 0;
}
int left = depth(root.left);
int right = depth(root.right);
return left>right?(left+1):(right+1);
}
其次是,采用两次递归,判断根左右子树是否为平衡二叉树:
public boolean IsBalanced_Solution(TreeNode root) {
if(root==null) return true;
if(Math.abs(getHeight(root.left)-getHeight(root.right))>1) return false;
return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);
}
public int getHeight(TreeNode root){
if(root==null) return 0;
return max(getHeight(root.left),getHeight(root.right))+1;
}
private int max(int a, int b){
return (a>b)?a:b;
}
但是这种做法很明显的一个问题就是,再判断上层结点的时候,会多次重复遍历下层结点,增加了不必要
的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平
衡二叉树,则直接停止遍历,这样至多只对每一个结点访问一次。
public boolean IsBalanced_Solution3(TreeNode root){
return etDepth(root) !=-1;
}
private int etDepth(TreeNode root){
if(root == null) return 0;
int left = etDepth(root.left);
if(left == -1) return -1;
int right = etDepth(root.right);
if(right == -1) return -1;
return Math.abs(left-right) >1 ? -1:1+Math.max(left, right);
}