57.剑指Offer-平衡二叉树

题目描述

平衡二叉树左右子树高度差不超过 1。

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

解题思路

private boolean isBalanced = true;

public boolean IsBalanced_Solution(TreeNode root) {
    height(root);
    return isBalanced;
}

private int height(TreeNode root) {
    if (root == null || !isBalanced)
        return 0;
    int left = height(root.left);
    int right = height(root.right);
    if (Math.abs(left - right) > 1)
        isBalanced = false;
    return 1 + Math.max(left, right);
}

就是在递归找每个子节点的高度，通过判断来给isBalanced赋值。