【LeetCode110】平衡二叉树

题目描述

给定一个二叉树,判断它是否是 平衡二叉树

思路与算法

二叉树是平衡的,如果对于每个节点,它的左子树和右子树之间的高度差最多为 1。

注意:根据上述定义,单独计算根节点的左右子树高度,然后进行高度比较,是不对的。

所以应该递归地确定每个子树的高度;同时,检查每个节点的平衡条件。(两个动作)

创建一个辅助函数(我们称之为 checkBalance )

  • 如果子树是平衡的,返回其高度
  • 如果是不平衡的,返回特殊值(比如-1)
  1. base case: 如果一个节点是 None ,那么它是一个高度为 0 的空子树。这被认为是平衡的。
  2. 递归步骤
    • 对于给定的节点,递归计算其左子树和右子树的高度。
    • 检查当前节点是否平衡
      • 如果左子树或右子树不平衡(即,递归调用返回 -1),那么当前节点的子树也不平衡------立即返回 -1。
      • 如果两个子树都是平衡的,那么检查它们高度之间的差异
      • 如果差值超过 1,则以当前节点为根的树是不平衡的;返回 -1
      • 否则,当前节点的高度是其左子树和右子树高度的最大值加 1。
  3. 结果整合
    Once the helper function completes for the root
    • 如果返回 -1,则树是不平衡的
    • 否则,树是平衡的

代码

python3 复制代码
# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def isBalanced(self, root: Optional[TreeNode]) -> bool:

        def checkBalance(node: TreeNode) -> int:
            # base case
            if not Node:
                return 0
            
            leftHight = checkBalance(node.left)
            if leftHight == -1:
                return -1
            
            rightHeight = checkBalance(node.right)
            if rightHeight == -1:
                return -1
            
            if abs(leftHight - rightHeight) > 1:
                return -1
            
            return 1 + max(leftHight, rightHeight)
        return checkBalance(root) != -1
        

总结与反思

  1. 将明确定义的递归步骤转换为代码或伪代码
    • 书写base case(s)
      • Clearly define the simplest case(s) where the answer is known
      • 在代码中,这通常以一个 if 语句的形式出现,该语句check base case 并立即返回一个值。
    • 如果f(n)是你的函数,展示f(n)是如何依赖f(n-1)或其它实例的
    • 通过arithmetic operations, concatenation, or some form of aggregation.合并子问题结果
  2. 何时考虑使用辅助函数进行代码实现?
    • 如果你的递归解决方案需要额外的信息(例如当前深度、边界或任何其他状态),而这些信息不属于原始函数的签名,那么使用辅助函数是个好主意。当你想要将复杂性隐藏在主接口之外,或者使代码更有组织和模块化时,也可以使用辅助函数。
      • 例如,在平衡二叉树问题中,辅助函数返回子树的高度和一个标志,指示子树是否平衡。这些额外的信息在原始函数签名中并未提供,因此将递归逻辑封装在辅助函数中可以保持主函数的接口简洁。
      • 在平衡树检查中,主函数 isBalanced 只需要知道辅助函数返回的是有效高度还是特殊值(如 -1),就可以判断是否平衡
  3. 递归实现时,哪些内容可以放入辅助函数?
    • 包括核心递归算法:基本情况、递归调用、子问题结果的组合,以及在递归过程中必须保持的任何额外状态或信息。
相关推荐
前端炒粉1 小时前
18.矩阵置零(原地算法)
javascript·线性代数·算法·矩阵
做运维的阿瑞2 小时前
用 Python 构建稳健的数据分析流水线
开发语言·python·数据分析
左师佑图2 小时前
综合案例:Python 数据处理——从Excel文件到数据分析
开发语言·python·数据分析·excel·pandas
暴风鱼划水2 小时前
三维重建【0-D】3D Gaussian Splatting:相机标定原理与步骤
算法·3d
l1t3 小时前
利用DeepSeek修改数据结构提升求解集合程序效率
数据结构·python·deepseek
jiushun_suanli3 小时前
PyTorch CV模型实战全流程(一)
人工智能·pytorch·python
2301_764441333 小时前
基于python构建的低温胁迫实验
开发语言·python
天才测试猿4 小时前
Selenium定位元素的方法css和xpath的区别
css·自动化测试·软件测试·python·selenium·测试工具·测试用例
云烟成雨TD4 小时前
NumPy 2.x 完全指南【四十二】线性代数之向量运算
python·机器学习·numpy