leetcode 110. 平衡二叉树

110. 平衡二叉树

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/balanced-binary-tree
著作权归领扣网络所有。

给定一个二叉树，判断它是否是高度平衡的二叉树。
本题中，一棵高度平衡二叉树定义为：

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

示例 1：

输入：root = [3,9,20,null,null,15,7]
输出：true

示例 2：

输入：root = [1,2,2,3,3,null,null,4,4]
输出：false

示例 3：

输入：root = []
输出：true

提示：

树中的节点数在范围 [0, 5000] 内
-104 <= Node.val <= 104

自顶向下-解法

空间复杂度 O(n^2)
空间复杂度 O(1)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int height(struct TreeNode * root) {
    if ( root == NULL ) {
        return 0;
    }

    return fmax( height(root->left), height(root->right) ) + 1;
}

bool isBalanced(struct TreeNode* root){
    if ( root == NULL ) {
        return true;
    }

    return fabs( height(root->left) - height(root->right) ) <= 1 && isBalanced(root->left) && isBalanced(root->right);
}

自底向上

空间复杂度 O(n)
空间复杂度 O(1)

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */

int height(struct TreeNode * root) {
    if ( root == NULL ) {
        return 0;
    }
    int left = height(root->left);
    int right = height(root->right);

    if ( left == -1 || right == -1 || fabs(left-right) > 1 ) {
        return -1;
    } else {
        return fmax(left, right) + 1;
    }
}

bool isBalanced(struct TreeNode* root) {
    return height(root) >= 0;
}