LeetCode 110. 平衡二叉树

给定一个二叉树，判断它是否是高度平衡的二叉树。

本题中，一棵高度平衡二叉树定义为：

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

示例 1:

给定二叉树 [3,9,20,null,null,15,7]
3
/ \
9 20
/ \
15 7
返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]
1
/ \
2 2
/ \
3 3
/ \
4 4
返回 false 。

tips : 自顶向上法, 时间复杂度O(n), 空间复杂度O(n)（递归栈）

class Solution {
public:
    bool isBalancedRecursive(TreeNode* root, int & height) {
        int left_height = 0;
        int right_height = 0;
        if(root == NULL) {
            return true;
        }
        if(isBalancedRecursive(root->left, left_height) && isBalancedRecursive(root->right, right_height) && abs(left_height - right_height) <= 1 ) {
            height = max(left_height, right_height)+1;
            return true;
        } else {
            height = max(left_height, right_height)+1;
            return false;
        }
    }
    bool isBalanced(TreeNode* root) {
        int height = 0;
        return isBalancedRecursive(root, height);
    }
};