首页 > 其他分享 >力扣 110. 平衡二叉树 [基础+优化]

力扣 110. 平衡二叉树 [基础+优化]

时间:2022-08-31 14:36:14浏览次数:80  
标签:right TreeNode cur 高度 110 力扣 二叉树 return left

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, 5000] 内
  • -104 <= Node.val <= 104

基础思路

题目要求每个节点 的左右两个子树的高度差的绝对值不超过 1,则每棵子数都要满足高度差的条件,构造一个函数getD(cur),计算以cur为根节点的树的最大高度,使用递归依次计算每棵树是否平衡,所有都满足则返回true,否则false

构造函数getD(cur)

getDepth伪代码
 int getDepth(TreeNode* cur){
        if(!cur)//当前节点为空
            return 0;
        if cur没有左子树 || cur没有右子树//只有cur这一个节点,则以cur为根节点的树高度为1
            return 1;
        if cur左右子树都存在//返回子数中最高的高度+1(1是cur这一层占一个高度)    
            return 1+max(getDepth(cur->left),getDepth(cur->right));
    }

优化

如果使用基础思路,会出现很多重复无用的计算,题目比如cur的左子树高度计算得到left已经不平衡,则无需计算右子树right的高度去比较是否平衡,也无需计算cur的高度。

因此优化在递归计算高度时,同时计算出是否满足平衡条件,left不仅表示左子树的高度,还增加-1的值表示左子树是否平衡,如果不平衡则节约right的计算和是否平衡的计算,直接将cur高度返回为-1right==-1同理,如果左右子数都平衡,则计算cur这棵树的高度进行返回。

例子如下图:

优化代码
 /**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode() : val(0), left(nullptr), right(nullptr) {}
 *     TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
 *     TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
 * };
 */
class Solution {
public:
    int getDepth(TreeNode* cur){
        if(!cur)//当前节点为空,则高度0
            return 0;
        int left=getDepth(cur->left);//获取左子数的高度
        if(left==-1)//左子树不平衡,则直接返回-1,表示cur这棵树不平衡
            return -1;
        int right=getDepth(cur->right);//获取右
        if(right==-1)//表示cur这棵树不平衡
            return -1;
        if(abs(left-right)>1)//计算cur这棵树是否平衡
            return -1;
        else
            return 1+max(left,right);//平衡,则返回cur这棵树的高度
    }
    bool isBalanced(TreeNode* root) {
        if(getDepth(root)==-1)
            return false;
        else
            return true;
    }
};

 

标签:right,TreeNode,cur,高度,110,力扣,二叉树,return,left
From: https://www.cnblogs.com/fudanxi/p/16642955.html

相关文章

  • 【数据结构】二叉树-二叉树类别
    满二叉树如果二叉树中除了叶子结点,每个结点的度都为2,则此二叉树称为满二叉树。 完全二叉树1.如果二叉树中除去最后一层节点为满二叉树,且最后一层的结点依次从左到右......
  • leetcode-998. 最大二叉树 II
    998.最大二叉树II图床:blogimg/刷题记录/leetcode/998/刷题代码汇总:https://www.cnblogs.com/geaming/p/16428234.html题目思路看到树就要想到递归。解法/***D......
  • 662. 二叉树最大宽度
    题目描述给你一棵二叉树的根节点root,返回树的最大宽度。树的最大宽度是所有层中最大的宽度。每一层的宽度被定义为该层最左和最右的非空节点(即,两个端点)之间......
  • 654.最大二叉树+998.最大二叉树II
    654.最大二叉树题目描述给定一个不重复的整数数组 nums。 最大二叉树 可以用下面的算法从 nums递归地构建:创建一个根节点,其值为 nums中的 最大值 。递......
  • 2022-8-30 每日一题-二叉树递归-
    998.最大二叉树II难度中等90收藏分享切换为英文接收动态反馈最大树 定义:一棵树,并满足:其中每个节点的值都大于其子树中的任何其他值。给你最大树的根节点 root......
  • 力扣372(java)-超级次方(中等)
    题目:你的任务是计算 ab 对 1337取模,a是一个正整数,b是一个非常大的正整数且会以数组形式给出。示例1:输入:a=2,b=[3]输出:8示例2:输入:a=2,b=[1,0]输出:102......
  • js 实现二叉树中序遍历
    varinorderTraversal=function(root){//迭代if(!root){return[];}letres=[];letstack=[];while(stack.length>......
  • 力扣50(java)-Pow(x,n)(中等)
    题目:实现 pow(x,n) ,即计算x的整数 n次幂函数(即xn)。 示例1:输入:x=2.00000,n=10输出:1024.00000示例2:输入:x=2.10000,n=3输出:9.26100示例3:输入:x=2......
  • js 实现二叉树前序遍历
    //前序遍历:根左右//中序遍历:左中右//后序遍历:左右根//迭代varpreorderTraversal=function(root){if(!root){returnnull;}//迭......
  • 重建二叉树
    题目链接:重建二叉树_牛客题霸_牛客网(nowcoder.com)根据前序遍历和中序遍历重建二叉树,返回根节点 importjava.util.*;/***Definitionforb......