LeetCode《程序员面试金典》面试题 04.04. 检查平衡性 & 剑指Offer 55 - II & 110 平衡二叉树

题目

解题

这道题中的平衡二叉树的定义是：二叉树的每个节点的左右子树的高度差的绝对值不超过 1，则二叉树是平衡二叉树。根据定义，一棵二叉树是平衡二叉树，当且仅当其所有子树也都是平衡二叉树，因此可以使用递归的方式判断二叉树是不是平衡二叉树，递归的顺序可以是自顶向下或者自底向上。

方法一：自顶向下的递归

有了计算节点高度的函数，即可判断二叉树是否平衡。具体做法类似于二叉树的前序遍历，即对于当前遍历到的节点，首先计算左右子树的高度，如果左右子树的高度差不超过 1，再分别递归地遍历左右子节点，并判断左子树和右子树是否平衡。

// javascript
var isBalanced = function(root) {
    if (root === null) return true;
    const heightL = getHeight(root.left);
    const heightR = getHeight(root.right);
    const heightDiff = Math.abs(heightL - heightR);
    if (heightDiff > 1) {
        return false;
    }
    else {
        // don't know whether its subtrees are balanced or not
        // [1,2,2,3,null,3,null,4,4,4,4]
        return isBalanced(root.left) && isBalanced(root.right);
    }
};

const getHeight = (root) => {
    if (root === null) return 0;
    return Math.max(getHeight(root.left), getHeight(root.right)) + 1;    
};

对于节点 $p$，如果它的高度是 $d$（根结点为0，下面每层+1），则 $height(p)$ 最多会被调用 $d$ 次（即遍历到它的每一个祖先节点时）。对于平均的情况，一棵树的高度 $h$ 满足 $O(h)=O(logn)$，因为 $d\le h$，所以总时间复杂度为 $O(nlogn)$。

空间复杂度：$O(h)$ ，其中 $h$ 为树的高度，空间复杂度主要取决于递归调用的层数，递归调用的层数不会超过 $h$。

方法二：自底向上的递归

方法一由于是自顶向下递归，因此对于同一个节点，函数 height 会被重复调用，导致时间复杂度较高。如果使用自底向上的做法，则对于每个节点，函数 height 只会被调用一次。

自底向上递归的做法类似于后序遍历，对于当前遍历到的节点，先递归地判断其左右子树是否平衡，再判断以当前节点为根的子树是否平衡。如果一棵子树是平衡的，则返回其高度（高度一定是非负整数），否则返回 Number.MIN_VALUE。如果存在一棵子树不平衡，则整个二叉树一定不平衡。

// javascript
var isBalanced = function(root) {
    return getHeight(root) !== Number.MIN_VALUE;
};

const getHeight = (root) => {
    if (root === null) return 0;
    const heightL = getHeight(root.left);
    const heightR = getHeight(root.right);
    const heightDiff = Math.abs(heightL - heightR);
    if (heightL === Number.MIN_VALUE || heightR === Number.MIN_VALUE || heightDiff > 1) {
        return Number.MIN_VALUE;
    }
    return Math.max(heightL, heightR) + 1;    
};

时间复杂度：$O(n)$ ，其中 $n$ 是二叉树中的节点个数。使用自底向上的递归，每个节点的计算高度和判断是否平衡都只需要处理一次，最坏情况下需要遍历二叉树中的所有节点。
空间复杂度：$O(h)$ ，其中 $h$ 为树的高度，空间复杂度主要取决于递归调用的层数，递归调用的层数不会超过 $h$。