lintcode--平衡二叉树

来源:互联网 发布:手机电影网php源码 编辑:程序博客网 时间:2024/06/10 03:13

给定一个二叉树,确定它是高度平衡的。对于这个问题,一棵高度平衡的二叉树的定义是:一棵二叉树中每个节点的两个子树的深度相差不会超过1。 

样例

给出二叉树 A={3,9,20,#,#,15,7}, B={3,#,20,15,7}

A)  3            B)    3    / \                  \  9  20                 20    /  \                / \   15   7              15  7

二叉树A是高度平衡的二叉树,但是B不是



/**
 * Definition of TreeNode:
 * public class TreeNode {
 *     public int val;
 *     public TreeNode left, right;
 *     public TreeNode(int val) {
 *         this.val = val;
 *         this.left = this.right = null;
 *     }
 * }
 */


public class Solution {
    /**
     * 先求出左右字树的最大深度(递归);
     * 再递归判断是否平衡
     */
   
    public int maxDepth(TreeNode root){
        if(root ==null){
            return 0;
        }
          int leftLen = 0 ;
        int rightLen = 0;
        leftLen = maxDepth(root.left);
        rightLen = maxDepth(root.right);
        return leftLen>rightLen?leftLen+1:rightLen+1;
    }
    public boolean isBalanced(TreeNode root) {
        // write your code here
        if(root == null){
            return true;
        }
        int leftLen = maxDepth(root.left);
        int rightLen = maxDepth(root.right);
        
        if((leftLen - rightLen)>1 || (rightLen-leftLen)>1){
            return false;
        }
        return isBalanced(root.left)&&isBalanced(root.right);
    }
}

原创粉丝点击