平衡二叉树-Java实现

平衡二叉树

http://www.nowcoder.com/questionTerminal/8b3b95850edb4115918ecebdf1b4d222

一. 思路

自底向上解法:按照平时做纸质考试的去思考如何判断一颗树是否平衡。采用后序遍历用自底向上递归方法,计算(左子树的深度-右子树的深度)的绝对值是否小于1,若是则平衡。

自顶向下解法:采用前序遍历一次树,计算每个节点的深度并存到map中。第二次遍历树,计算高度差是否符合平衡的要求

二. 递归解法的代码

public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        return depthOfNode(root) != -1;
    }

    public int depthOfNode(TreeNode root) {
        if (root == null) return 0;
        int leftDepth = depthOfNode(root.left);
        if (leftDepth == -1) return -1; 
        int rightDepth = depthOfNode(root.right);
        if (rightDepth == -1) return -1;
        if ((leftDepth - rightDepth) < -1 || (leftDepth -rightDepth) > 1) {
            return -1;
        } else {
            return 1 + (leftDepth > rightDepth ? leftDepth : rightDepth);
        }

    }
}
全部评论

相关推荐

喜欢核冬天的哈基米很...:会爆NullPointerException的
点赞 评论 收藏
分享
粗心的熊熊求求off...:什么内容都没有还弄两页
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务