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

剑指Offer-- 平衡二叉树

时间:2018-07-26 19:58:06      阅读:106      评论:0      收藏:0      [点我收藏+]

标签:ack   root   min   平衡   访问   左右   tree   左右子树   停止   

题目 输入一棵二叉树,判断该二叉树是否是平衡二叉树。
思路 递归判断,先判断自身左右子树高度差,再递归判断左右子树是否为平衡二叉树;

package solution;

public class Solution28 {
    
     public boolean IsBalanced_Solution(TreeNode root) {
            if(root==null) { //递归出口;
                return true;
            }
           
            return (Math.abs(getTreeHeight(root.left)-getTreeHeight(root.right))<=1)&&IsBalanced_Solution(root.left)&&IsBalanced_Solution(root.right);
        }
        private int getTreeHeight(TreeNode root) {
            if (root == null) {
                return 0;
            }
            return 1 + Math.max(getTreeHeight(root.left), getTreeHeight(root.right));
        }
}
            

这种做法有很明显的问题,在判断上层结点的时候,会多次重复遍历下层结点,增加了不必要的开销。如果改为从下往上遍历,如果子树是平衡二叉树,则返回子树的高度;如果发现子树不是平衡二叉树,则直接停止遍历,这样至多只对每个结点访问一次。
更巧妙的做法 求高度的同时判断是否为平衡二叉树

链接:https://www.nowcoder.com/questionTerminal/8b3b95850edb4115918ecebdf1b4d222
来源:牛客网

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root) != -1;
    }
     
    private 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);
    }
}

剑指Offer-- 平衡二叉树

标签:ack   root   min   平衡   访问   左右   tree   左右子树   停止   

原文地址:https://www.cnblogs.com/LynnMin/p/9373750.html

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