C++ 之LeetCode刷题记录（二十）

开始cpp刷题之旅。

依旧是追求耗时0s的一天。

110. 平衡二叉树

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

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

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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 ；

当左右子树深度差>2时，返回-1；

当左右子树深度差<=1时，返回左右子树之间大的值再加上根节点1，即为树的深度；

最后当节点为空，或者左右节点的值为-1，即不平衡树时，停止递归；

通过判断返回值是否为-1，确定是否为平衡树。

class Solution {
public:
    bool isBalanced(TreeNode* root) {
        return recur(root) != -1;
    }
private:
    int recur(TreeNode* root) {
        if (root == nullptr) return 0;
        int left = recur(root->left);
        if (left == -1) return -1;
        int right = recur(root->right);
        if (right == -1) return -1;
        return abs(left - right) < 2 ? max(left, right) + 1 : -1;
    }
};