【算法练习】判断是不是平衡二叉树

题源：

牛客

描述

输入一棵节点数为 n 二叉树，判断该二叉树是否是平衡二叉树。

在这里，我们只需要考虑其平衡性，不需要考虑其是不是排序二叉树

平衡二叉树（Balanced Binary Tree），具有以下性质：它是一棵空树或它的左右两个子树的高度差的绝对值不超过1，并且左右两个子树都是一棵平衡二叉树。

样例解释：

样例二叉树如图，为一颗平衡二叉树

注：我们约定空树是平衡二叉树。

数据范围：n≤100,树上节点的val值满足0≤n≤1000

要求：空间复杂度O(1)，时间复杂度O(n)

输入描述：

输入一棵二叉树的根节点

返回值描述：

输出一个布尔类型的值

示例1

输入：

{1,2,3,4,5,6,7}

返回值：

true

示例2

输入：

{}

返回值：

true

思路：

先写函数求子树高度，然后将两边高度相减求，如果超过了1那么就不平衡，返回错误；如果没有超过，则继续判断接下来的左右子树。

题解：

class Solution {
public:
    int maxDepth(TreeNode* root){
        if(root == nullptr) return 0;
        //求子树高度
        return 1 + max(maxDepth(root->left),maxDepth(root->right));
    }
    bool IsBalanced_Solution(TreeNode* pRoot) {
        //如果是空树，直接返回true
        if(pRoot == nullptr)    return true;
        //左右子树高度
        int left = maxDepth(pRoot->left);
        int right = maxDepth(pRoot->right);
        //如果超过1了，就不平衡，返回错误
        if(abs(left - right) > 1)   return false;
        //递归判断接下来的左右子树
        return IsBalanced_Solution(pRoot->left) && IsBalanced_Solution(pRoot->right);
    }
};

优化版

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