平衡二叉树

平衡二叉树

描述

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

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

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

示例 1:

给定二叉树 [3,9,20,null,null,15,7]

3

/
9 20
/
15 7
返回 true 。

思路和代码

首先看递归求二叉树高度的函数

int height(TreeNode* root)
    {
        if(root == NULL)
            return 0;
        int left = height(root->left);
        int right = height(root->right);
        return left > right ? left + 1 : right + 1;
    }

再看看主函数

bool isBalanced(TreeNode* root) {
        if(root == NULL)
            return true;
        int left = height(root->left);
        int right = height(root->right);
        if(abs(left-right) > 1)
            return false;
        return isBalanced(root->left) && isBalanced(root->right);
    }