day21-110.平衡二叉树

110.平衡二叉树

力扣题目链接

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

本题中,一棵高度平衡二叉树定义为:一个二叉树每个节点 的左右两个子树的高度差的绝对值不超过1。

示例 1:

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

返回 true 。

示例 2:

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

返回 false 。

知识准备

这里要强调一些概念,什么是深度又什么是高度。

  • 深度:根节点到叶子结点的距离
  • 高度:叶子节点到根节点的距离

用一张图会更加直观

对于深度和高度的求法,各有不同,求深度应该从上到下,使用前序遍历 ,而高度是从下到上,应该是后序遍历

在之前的求二叉树的最大深度时,使用的是后序,原因是求的是叶子节点的深度,也为根节点的高度。

如果是正经用前序遍历求深度,代码如下:

cpp 复制代码
class Solution {
private:
    int res;
public:
    void getDepth(TreeNode* node, int depth){
        res = max(res,depth);
        if(node->left == nullptr && node->right == nullptr)return ;

        if(node->left){
            depth++;
            getDepth(node->left,depth);
            depth--;
        }
        if(node->right){
            depth++;
            getDepth(node->right,depth);
            depth--;
        }
        return;
    }
    int maxDepth(TreeNode* root) {
        res = 0;
        if(root == nullptr)return res;
        getDepth(root,1);
        return res;
    }
};

思路

首先确定我们要求的应该是高度,使用后序遍历。

  • 确定终止条件

if(root == null)return 0

返回0则是高度为0

  • 单层循环

确定终止条件后,分别求左右子树的高度。

cpp 复制代码
		int leftHeight = getHeight(root->left);
        if(leftHeight == -1)return -1;
        int rightHeight = getHeight(root->right);
        if(rightHeight == -1)return -1;
  • 确定高度差
cpp 复制代码
		int res = 0;
        res = abs(leftHeight-rightHeight) > 1?-1:1+max(leftHeight,rightHeight);
        return res;

最终代码如下:

cpp 复制代码
class Solution {
public:
    int getHeight(TreeNode* root){
        if(root == nullptr)return 0;
        
        int leftHeight = getHeight(root->left);
        if(leftHeight == -1)return -1;
        int rightHeight = getHeight(root->right);
        if(rightHeight == -1)return -1;

        int res = 0;
        res = abs(leftHeight-rightHeight) > 1?-1:1+max(leftHeight,rightHeight);
        return res;
    }
    bool isBalanced(TreeNode* root) {
        return getHeight(root) == -1?false:true;
    }
};

eturn res;

}

bool isBalanced(TreeNode* root) {

return getHeight(root) == -1?false:true;

}

};

复制代码
相关推荐
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】两次二分查找
开发语言·数据结构·python·算法·leetcode
秋说3 小时前
【PTA数据结构 | C语言版】一元多项式求导
c语言·数据结构·算法
谭林杰4 小时前
B树和B+树
数据结构·b树
卡卡卡卡罗特5 小时前
每日mysql
数据结构·算法
chao_7896 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
lifallen6 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
不吃洋葱.7 小时前
前缀和|差分
数据结构·算法
哦吼!9 小时前
数据结构—二叉树(二)
数据结构
码农Cloudy.11 小时前
C语言<数据结构-链表>
c语言·数据结构·链表
lightqjx11 小时前
【数据结构】顺序表(sequential list)
c语言·开发语言·数据结构·算法