【Java】 劍指offer(55-2) 平衡二叉樹

 

本文參考自《劍指offer》一書，代碼採用Java語言。

更多：《劍指Offer》Java實現合集  

題目

　　輸入一棵二叉樹的根結點，判斷該樹是否是平衡二叉樹。若是某二叉樹中任意結點的左右子樹的深度相差不超過1，那麼它就是一棵平衡二叉樹。

思路

　　在(55-1) 二叉樹的深度基礎上修改：計算樹的深度，樹的深度=max(左子樹深度，右子樹深度)+1。在遍歷過程當中，判斷左右子樹深度相差是否超過1，若是不平衡，則令樹的深度=-1，用來表示樹不平衡。最終根據樹的深度是否等於-1來肯定是否爲平衡樹。

 

測試算例　

　　1.功能測試（左斜樹、右斜樹、平衡或者不平衡樹）

　　3.特殊測試（一個結點，null）

Java代碼

//題目：輸入一棵二叉樹的根結點，判斷該樹是否是平衡二叉樹。若是某二叉樹中
//任意結點的左右子樹的深度相差不超過1，那麼它就是一棵平衡二叉樹。

public class BalancedBinaryTree {
	public class TreeNode {
	    int val = 0;
	    TreeNode left = null;
	    TreeNode right = null;

	    public TreeNode(int val) {
	        this.val = val;
	    }
	}
	
    public boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root)!=-1;
    }
    
    public int getDepth(TreeNode root) {
        if(root==null)    return 0;
        int left=getDepth(root.left);
        if(left==-1)    return -1;
        int right=getDepth(root.right);
        if(right==-1)    return -1;
        return Math.abs(left - right) > 1 ? -1 : 1 + Math.max(left, right);
    }
    
    /*
    //本身開始想的方法，可是必定要把樹給遍歷完才行；上面的方法實現了剪枝
    boolean isBalanced=true;
    public boolean IsBalanced_Solution(TreeNode root) {
        TreeDepth(root);
        return isBalanced;
    }
     
    public int TreeDepth(TreeNode root) {
        if(root==null)
            return 0;
        int left=TreeDepth(root.left);
        int right=TreeDepth(root.right);
        if(left-right>1 || right-left>1)
            isBalanced=false;
        return Math.max(left+1,right+1);
    }
    */
}

　　

收穫

　　1.在判斷出樹不平衡後，進行剪枝（即代碼中直接返回-1，再也不對其餘子樹進行判斷），以提升效率。

　　

更多：《劍指Offer》Java實現合集