码迷,mamicode.com
首页 > 其他好文 > 详细

LeetCode-Balanced Binary Tree

时间:2015-08-14 19:18:52      阅读:123      评论:0      收藏:0      [点我收藏+]

标签:

Given a binary tree, determine if it is height-balanced.

For this problem, a height-balanced binary tree is defined as a binary tree in which the depth of the two subtrees of every node never differ by more than 1.

感觉树需要很多工具类的方法,其实都是改造出来的,如果做这道题就方便了,求树的高度,可以看之前的blog。http://blog.csdn.net/my_jobs/article/details/47662437 所以,直接按题意改造就可以了,上代码:

    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        if (Math.abs(maxDepth(root.left)-maxDepth(root.right)) > 1) return false;
        return isBalanced(root.left) && isBalanced(root.right);
    }
    
    private int maxDepth(TreeNode root) {
        if (root == null) return 0;
        return Math.max(maxDepth(root.left), maxDepth(root.right))+1;
    }

也是很精炼的代码,但是明显时间复杂度比较高。所以现在就可以改造代码,在求depth的时候就判断是不是balance的。

    public boolean isBalanced(TreeNode root) {
        if (root == null) return true;
        if (depth(root) == -1) return false;
        return true;
    }
    
    private int depth(TreeNode root) {
        if (root == null) return 0;
        int left = depth(root.left);
        int right = depth(root.right);
        if (left == -1 || right == -1) return -1;
        if (Math.abs(left-right) > 1) return -1;
        return Math.max(left, right)+1;
    }

第一种方法有点直译的感觉,是一个自上至下的顺序;但是第二种是一个自下向上的顺序。因为计算高度也是从下向上的过程,所以可以顺便优化算法!

其实,树的这种递推性质,如果可以自下向上都是对之前算法的一个很好的优化!!!多感悟此类问题。



版权声明:本文为博主原创文章,未经博主允许不得转载。

LeetCode-Balanced Binary Tree

标签:

原文地址:http://blog.csdn.net/my_jobs/article/details/47663845

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