LeetCode 剑指 Offer 55 - II. 平衡二叉树 LCOF

LeetCode 剑指 Offer 55 - II. 平衡二叉树 LCOF

题目描述

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

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

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

示例 1:

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

返回 true 。

示例 2:

给定二叉树 [1,2,2,3,3,null,null,4,4]

返回 false 。

解题

    可以通过迭代法获取每个节点到叶节点的最大高度，因此可以比较当前节点左右子树的高度，我们只需要在迭代过程中比较左右子树的高度差就能判断是否是平衡二叉树，因此需要设计一个迭代函数，它能返回当前节点对应子树的高度，也能表明子树是否满足平衡二叉树要求，那么如果当前子树满足平衡二叉树要求，则返回其高度，否则返回-1，因此如果其子树返回值中存在-1或两子树高度差大于1，则该子树返回-1。

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    int getH(TreeNode *root){
        if (root == NULL) return 0;
        int lh = getH(root->left);
        int rh = getH(root->right);
        if (lh > rh) swap(lh, rh);
        if (lh==-1 || rh==-1 || rh - lh > 1) return -1;
        return rh+1;
    }
    bool isBalanced(TreeNode* root) {
        return getH(root) != -1;
    }
};

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    bool isBalanced(TreeNode* root) {
        int h = getH(root);
        return isBalance;
    }
private:
    bool isBalance = true;
    
    int getH(TreeNode *root){
        if (root == NULL) return 0;
        int lh = getH(root->left);
        int rh = getH(root->right);
        if (lh==-1 || rh==-1 || abs(rh - lh) > 1) isBalance = false;
        return max(rh,lh)+1;
    }
};