leetcode 110-判断一棵树是否为平衡二叉树 #算法#

原题如下

Given a binary tree, determine if it is height-balanced.
给出一棵二叉树，判断它是否高度平衡。
For this problem, a height-balanced binary tree is defined as:
高度平衡二叉树定义为：

a binary tree in which the depth of the two subtrees of every node never differ by more than 1.
任何节点的两棵子树的深度差不能大于一。

换个说法就是，左右子树的高度（深度）差不能超过一并且左右子树也是平衡二叉树。

思路

首先需要计算一棵树的深度，一棵树的深度为其左右子树的较大深度加上一，左右子树的深度也可以用同样的方法计算出，可以用递归实现，终止条件是根节点为空时深度为0；
有了深度之后，就可以比较某一节点的左右子树的深度差是否小于等于1，并要求左右子树也是平衡树，同样可以用递归实现，终止条件是空树是平衡树。

代码

class Solution {
public:
    int depth(TreeNode* root){
        if(root == NULL) return 0;
        int leftDepth = depth(root->left);
        int rightDepth = depth(root->right);
        return (leftDepth > rightDepth ? leftDepth : rightDepth) + 1;
    }
    
    bool isBalanced(TreeNode* root) {
        if(root == NULL) return true;
        return abs(depth(root->left) - depth(root->right)) <= 1 
            && isBalanced(root->left) 
            && isBalanced(root->right);
    }
};