【力扣面试】面试题 04.04. 检查平衡性

【力扣面试】面试题 04.04. 检查平衡性

题目

实现一个函数，检查二叉树是否平衡。在这个问题中，平衡树的定义如下：任意一个节点，其两棵子树的高度差不超过 1。

示例 1:

示例 2:

解题思路

1、分别求取当前节点的左子树高度和右子树高度，比较其差值是否小于1
2、并且判断当前节点的左孩子节点是否满足1条件
3、并且判断当前节点的右孩子节点是否满足1条件
4、都满足的话，即可返回true

代码

class Solution {
    public boolean isBalanced(TreeNode root) {
        if (root==null){
            return true;
        }
        //分别计算左子树和右子树的高度
        int left = getDepth(root.left);
        int right = getDepth(root.right);
        //这两个子树的高度不能超过1，并且他的两个子树也必须是平衡二叉树
        return Math.abs(left-right) <= 1 && isBalanced(root.left) && isBalanced(root.right);
    }
    //计算树中节点的高度
    public int getDepth(TreeNode root){
        if (root == null) {
            return 0;
        }
        return (Math.max(getDepth(root.left), getDepth(root.right)) + 1);
    }
}