平衡二叉树-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);
        }

    }
}
全部评论

相关推荐

牛客101244697号:这个衣服和发型不去投偶像练习生?
点赞 评论 收藏
分享
11-07 13:31
怀化学院 Java
勇敢牛牛不怕难:又疯一个
点赞 评论 收藏
分享
11-28 17:48
中山大学 C++
点赞 评论 收藏
分享
评论
点赞
收藏
分享
牛客网
牛客企业服务