如何判断一个二叉树是否是平衡二叉树(Java)

假如有一个二叉树,需要确定它是否是平衡二叉树。最直接的做法,就是遍历每个结点,借助一个获取树深度的递归函数,根据该结点的左右子树高度差判断是否平衡,然后递归地对左右子树进行判断。

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);
	 }

 

    原文作者:平衡二叉树
    原文地址: https://blog.csdn.net/qq_23031939/article/details/82120844
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞