【Leetcode二叉树属性五】110. 平衡二叉树程序员面试金典-面试题04.04.检查平衡性

本文链接：https://blog.csdn.net/qq_46514118/article/details/120596545

文章目录

Leetcode110，面试题04.04

Leetcode110，面试题04.04

1.问题描述

在这里插入图片描述

2.解决方案

不管是下面哪种解决方案核心抓住解决问题的两个步骤
1.检查以root为根的树是否为平衡树即：abs(getHeight(root->left)-getHeight(root->right))<=1
2.检查root的两颗子树是否为平衡树即：
左子树：abs(getHeight(root->left->left)-getHeight(root->left->right))<=1
右子树：abs(getHeight(root->right->left)-getHeight(root->right->right))<=1
3.检查一棵树的平衡性==以root为根的树是平衡树&&root的两棵子树为平衡树

解法一：错误低效递归

错误在于：你只判断了根的左右子树高度差不超过一但是这不能保证左右子树等孙子树都保证

//递归错误写法
class Solution {
public:
    int getHeight(TreeNode* root){
        //空树高度为-1 单节点树高度0
        if(root==nullptr) return -1;
        return max(getHeight(root->right),getHeight(root->left))+1;

    }
    //错误在于 你只判断了根的左右子树高度差不超过一 但是这不能保证左右子树等孙子树都保证
    bool isBalanced(TreeNode* root) {
        if(root==nullptr||(root->left==nullptr&&root->right==nullptr)) return true;

        if(abs(getHeight(root->right)-getHeight(root->left))<=1) return true;
        return false;
    }
};

解法二：正确低效递归

思路就是对于每一个结点都获取他的高度，然后分两步先判断自己再判断左右子树
1.相当于判断完自己了以当前节点为根节点的树是正确的
2.判断为自己也即当前节点就得判断左右子树开始递归

//递归正确写法
class Solution1 {
public:
    int getHeight(TreeNode* root){
        //空树高度为-1 单节点树高度0
        if(root==nullptr) return -1;
        return max(getHeight(root->right),getHeight(root->left))+1;

    }

    bool isBalanced(TreeNode* root) {
        //相当于判断完自己了 以当前节点为根节点的树是正确的
        if(root==nullptr||(root->left==nullptr&&root->right==nullptr)) return true;
        if(abs(getHeight(root->right)-getHeight(root->left))>1) return false;

        //判断为自己也即当前节点 就得判断左右子树 开始递归
        return isBalanced(root->left)&&isBalanced(root->right);
    }
};

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

解法三：正确高效递归

大致思路：
1.这个程序员面试金典上面有详细解释，就是上面的低效递归有很多对于节点高度的重复计算
2.这个算法只要出现以某个结点为根的树不平衡会立刻将错误信息return，那么它的父节点等等递归调用者收到这个消息就会立刻继续return上去

总体也分为两步大概：
1.递归调用：相当于检查root的两颗子树是不是平衡树
2.本层逻辑：相当于检查root为根的树是不是平衡树

//递归更高效正确写法
class Solution2 {
public:
    int checkHeight(TreeNode* root){
        if(root==nullptr) return -1;
		
		//1.相当于检查root的两颗子树是不是平衡树
        //递归获得子树的高度
        //并且判断是否有从下面传上来的错误信息,有的话直接return
        int leftHeight=checkHeight(root->left);
        if(leftHeight==INT16_MIN) return INT16_MIN;
        int rightHeight=checkHeight(root->right);
        if(rightHeight==INT16_MIN) return INT16_MIN;
		
		//2.相当于检查root为根的树是不是平衡树
        //检查如果两个子树的高度相差不超过一,那以root为根的树是平衡的直接返回高度
        //如果不是那证明以root为根的树不是平衡的那就返回错误值
        if(abs(leftHeight-rightHeight)>1) return INT16_MIN;
        else return max(leftHeight,rightHeight)+1;

    }

    bool isBalanced(TreeNode* root) {
        if(checkHeight(root)==INT16_MIN) return false;
        else return true;
    }
};