【剑指OFFER】输入一棵二叉树，判断该二叉树是否是平衡二叉树

题目描述

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

分析

平衡二叉树：一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

public boolean IsBalanced_Solution(TreeNode root) {
    //判断是否为空树
    if (root == null) {
        return true;
    }
    //左右两个子树的高度差的绝对值不超过1
    int leftDepth = getTreeDepth(root.left);
    int rightDepth = getTreeDepth(root.right);
    if (Math.abs(leftDepth - rightDepth) > 1) {
        return false;
    }
    //左右两个子树都是一棵平衡二叉树
    return IsBalanced_Solution(root.left) && IsBalanced_Solution(root.right);

}

//得到树的高度
private int getTreeDepth(TreeNode root) {
    if (root == null) {
        return 0;
    }
    int leftDepth = getTreeDepth(root.left);
    int rightDepth = getTreeDepth(root.right);
    return Math.max(leftDepth, rightDepth) + 1;
}