剑指Offer——平衡二叉树

最新推荐文章于 2022-07-10 12:02:48 发布

原创最新推荐文章于 2022-07-10 12:02:48 发布 · 148 阅读

0 ·

CC 4.0 BY-SA版权

剑指Offer 专栏收录该内容

65 篇文章

订阅专栏

本文介绍了判断二叉树是否为平衡二叉树的两种方法，一种是递归遍历，另一种是从下往上遍历，使用后续遍历左右根，记录左右子树深度，避免重复计算。提供了具体的代码实现。

题目：输入一棵二叉树，判断该二叉树是否是平衡二叉树。

定义：如果某二叉树中任意节点的左右子树的深度相差不超过1，那么它就是一颗平衡二叉树。

思路一：递归重复遍历左右子树，类似于求二叉树的深度；见《剑指Offer——二叉树的深度》

此思路判断上层节点的时候，会多次重复下层节点，增加了不必要的开销；

思路二：改为从下往上遍历，使用后续遍历左右根，遍历的时候记录左右子树的深度；

如果是平衡二叉树返回子树的高度，如果不是，则停止遍历；

代码：

public static boolean IsBalanced_Solution(TreeNode root) {
        return getDepth(root) != -1;
    }

    private static 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);
    }

参考链接：牛客网-丁满历险记的回答