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

    }
}
全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务