剑指offer — 平衡二叉树

判断二叉树是否平衡

最新推荐文章于 2021-07-25 16:20:34 发布

原创最新推荐文章于 2021-07-25 16:20:34 发布 · 332 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#path #标签 #class

剑指offer 专栏收录该内容

64 篇文章

订阅专栏

本文介绍了一种判断二叉树是否为平衡二叉树的方法。通过递归地计算每个节点的左右子树深度，并检查深度差是否超过1来确定整棵树是否平衡。

输入一棵二叉树，判断该二叉树是否是平衡二叉树

java

class Tree {
    boolean isBalance;
    int deep;
    public Tree(boolean isBalance, int deep) {
        this.isBalance = isBalance;
        this.deep = deep;
    }
}
public class Solution {
    public boolean IsBalanced_Solution(TreeNode root) {
        if (root == null) {
            return true;
        }
        if (root.left == null && root.right == null) {
            return true;
        }
        return isBal(root).isBalance;
    }
    private Tree isBal(TreeNode root) {
        if (root == null) {
            return new Tree(true, 0);
        }
        if (root.left == null && root.right == null) {
            return new Tree(true, 1);
        }
        Tree left = isBal(root.left);
        Tree right = isBal(root.right);
        if (!left.isBalance || !right.isBalance) {
            return new Tree(false, -1);
        }
        if (Math.abs(left.deep - right.deep) > 1) {
            return new Tree(false, -1);
        }
        return new Tree(true, Math.max(left.deep, right.deep) + 1);
    }
}