【C语言刷LeetCode】110. 平衡二叉树(E)

【

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

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

一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过 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, 5000] 内
-104 <= Node.val <= 104

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/balanced-binary-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

】

题目需要算一下深度，所以需要从叶子节点算起，很多题都是用到这种思想，所以这道简单题也记录下。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     struct TreeNode *left;
 *     struct TreeNode *right;
 * };
 */


int Preorder(struct TreeNode* root)
{
    if (root == NULL) { // 其实甚至是从叶子节点开始向上算
        return 1;
    }

    int lcnt = Preorder(root->left);
    int rcnt = Preorder(root->right);

    if (lcnt == -1 || rcnt == -1) {
        return -1;
    }

    int min = abs(lcnt - rcnt);
    if (min > 1) {
        return -1;
    }

    return fmax(lcnt, rcnt) + 1;
}

bool isBalanced(struct TreeNode* root){
    if (root == NULL) {
        return true;
    }

    int rcnt = Preorder(root);  // 返回树的深度

    if (rcnt == -1) { // -1表示返回的false
        return false;
    } else {
        return true;
    }
}