码迷,mamicode.com
首页 > 编程语言 > 详细

【Java】 剑指offer(55-2) 平衡二叉树

时间:2018-11-14 19:04:56      阅读:123      评论:0      收藏:0      [点我收藏+]

标签:判断   表示   balance   mat   https   turn   收获   语言   过程   

 

本文参考自《剑指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实现合集  

【Java】 剑指offer(55-2) 平衡二叉树

标签:判断   表示   balance   mat   https   turn   收获   语言   过程   

原文地址:https://www.cnblogs.com/yongh/p/9959131.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!