LeetCode - 110. 平衡二叉树

思路

一棵高度平衡二叉树定义为：

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

1. 定一个辅助方法，求二叉树的高度。
   递归：
   • 如果根节点为 null，返回 0.
   • 如果根节点不为 null，返回左、右子树二者中最高的高度 + 1。
2. 直接递归检查每个节点的左右两个子树的高度差是否小于等于 1。

复杂度分析

假设树有 n 个节点。

• 时间复杂度O(nlogn)，遍历两遍整棵树（检查高度差遍历一遍，同时计算深度遍历一遍）。
• 空间复杂度O(nlogn)，空间复杂度取决二叉树的节点数。

代码

class Solution {
    public static boolean isBalanced(TreeNode root) {
        if (root == null){
            return true;
        }
        if (Math.abs(depthOfTree(root.left) - depthOfTree(root.right)) > 1){
            return false;
        }
        return isBalanced(root.left) && isBalanced(root.right);
    }

    private static int depthOfTree(TreeNode root) {
        if (root == null) {
            return 0;
        }
        return Math.max(depthOfTree(root.left), depthOfTree(root.right)) + 1;
    }
}