美文网首页
LeetCode 110. Balanced Binary Tr

LeetCode 110. Balanced Binary Tr

作者: _Zy | 来源:发表于2021-03-31 15:48 被阅读0次

    Leetcode : Balanced Binary Tree
    Leetcode中文站:平衡二叉树
    Diffculty:Easy

    关键字:Tree,递归

    描述:
    判断一棵树是否是高度平衡的二叉树。
    在本题中,高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 1

    思路:
    其实思路很简单,使用递归可以很容易写出。
    1、判断左子树是否平衡
    2、判断右子树书否平衡
    3、判断左右子树高度差是否小于等于1.(当前节点为root是否平衡)

    要注意的点:
    在Leetcode中文站官方解中,给出了两种方式,自顶向下和自底向上。但这两种方式都不可避免的进行了多余的递归和多余的计算。
    1、判断平衡 和 计算高度。这两个操作都会对树进行递归。如果分开操作,那么会有两次递归。
    2、如果有某一个子树不平衡,那么整个树就不平衡,可以直接返回结果。而不需要再去计算其他子树了。

    我的思路:
    就是解决上面两点的问题。
    1、在判断子树平衡的同时,计算树的高度。如果平衡则返回高度,不平衡返回-1。
    2、如果子树不平衡直接快速失败,避免重复计算。

    运行结果:
    时间:1ms, beats 99.97%
    空间:38.4MB, beats 82.19%

        public boolean isBalanced(TreeNode root) {
            if(root == null) return true;
            return balanceDepth(root, 1) > 0;
        }
    
        /**
         * 思路:
         * 1、一次递归。在判断子树平衡的同时返回高度。如果不平衡则返回-1
         * 2、快速失败。当判断到某一个子树不平衡时直接返回失败,不再往下递归。
         * @param node
         * @param depth
         * @return
         */
        private int balanceDepth(TreeNode node, int depth){
            if(node == null){
                return depth;
            }
            int leftDepth = balanceDepth(node.left, depth);
            if(leftDepth < 0) {
                return -1;
            }
            int rightDepth = balanceDepth(node.right, depth);
            if(rightDepth < 0){
                return -1;
            }
            if(Math.abs(leftDepth-rightDepth) <=1){
                return depth + Math.max(leftDepth, rightDepth);
            }
            return -1;
        }
    

    相关文章

      网友评论

          本文标题:LeetCode 110. Balanced Binary Tr

          本文链接:https://www.haomeiwen.com/subject/xhxahltx.html