LeetCode 热题 100 543. 二叉树的直径

LeetCode 热题 100 | 543. 二叉树的直径

大家好,今天我们来解决一道经典的二叉树问题------二叉树的直径。这道题在 LeetCode 上被标记为简单难度,要求计算给定二叉树的直径。


问题描述

给你一棵二叉树的根节点,返回该树的直径。二叉树的直径是指树中任意两个节点之间最长路径的长度。这条路径可能经过也可能不经过根节点。

示例 1:

plaintext 复制代码
输入:root = [1,2,3,4,5]
输出:3
解释:3,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。

示例 2:

plaintext 复制代码
输入:root = [1,2]
输出:1

提示:

  • 树中节点数目在范围 [1, 10^4]
  • -100 <= Node.val <= 100

解题思路

核心思想
  1. 递归法

    • 使用递归法计算每个节点的左子树和右子树的最大深度。
    • 对于每个节点,计算其左子树深度和右子树深度之和,作为该节点的直径。
    • 更新全局最大直径。
  2. 辅助函数

    • 定义一个辅助函数 maxDepth,用于计算以某个节点为根的子树的最大深度。

Python代码实现

python 复制代码
class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        self.max_diameter = 0
        
        def maxDepth(node: TreeNode) -> int:
            if not node:
                return 0
            left_depth = maxDepth(node.left)
            right_depth = maxDepth(node.right)
            # 更新全局最大直径
            self.max_diameter = max(self.max_diameter, left_depth + right_depth)
            return max(left_depth, right_depth) + 1
        
        maxDepth(root)
        return self.max_diameter

代码解析

  1. 初始化

    • 定义一个全局变量 self.max_diameter,用于存储全局最大直径。
  2. 递归函数

    • 定义一个辅助函数 maxDepth,用于计算以某个节点为根的子树的最大深度。
    • 如果节点为空,返回 0。
    • 递归计算左子树和右子树的最大深度。
    • 更新全局最大直径为左子树深度和右子树深度之和。
    • 返回当前节点的最大深度。
  3. 主函数

    • 调用 maxDepth 函数,从根节点开始计算最大深度。
    • 返回全局最大直径。

复杂度分析

  • 时间复杂度 :O(n),其中 n 是树中节点的数量。每个节点被访问一次。
  • 空间复杂度 :O(h),其中 h 是树的高度。递归调用栈的深度最多为树的高度。

示例运行

示例 1
plaintext 复制代码
输入:root = [1,2,3,4,5]
输出:3
解释:3,取路径 [4,2,1,3] 或 [5,2,1,3] 的长度。
示例 2
plaintext 复制代码
输入:root = [1,2]
输出:1

总结

通过递归法,我们可以高效地计算二叉树的直径。递归函数 maxDepth 用于计算每个节点的左子树和右子树的最大深度,并更新全局最大直径。希望这篇题解对大家有所帮助,如果有任何问题,欢迎在评论区留言讨论!

关注我,获取更多算法题解和编程技巧!

相关推荐
我命由我1234512 分钟前
Kotlin 数据容器 - List(List 概述、创建 List、List 核心特性、List 元素访问、List 遍历)
java·开发语言·jvm·windows·java-ee·kotlin·list
励志要当大牛的小白菜2 小时前
ART配对软件使用
开发语言·c++·qt·算法
qq_513970442 小时前
力扣 hot100 Day56
算法·leetcode
武子康2 小时前
Java-80 深入浅出 RPC Dubbo 动态服务降级:从雪崩防护到配置中心秒级生效
java·分布式·后端·spring·微服务·rpc·dubbo
PAK向日葵3 小时前
【算法导论】如何攻克一道Hard难度的LeetCode题?以「寻找两个正序数组的中位数」为例
c++·算法·面试
爱喝矿泉水的猛男5 小时前
非定长滑动窗口(持续更新)
算法·leetcode·职场和发展
YuTaoShao5 小时前
【LeetCode 热题 100】131. 分割回文串——回溯
java·算法·leetcode·深度优先
源码_V_saaskw5 小时前
JAVA图文短视频交友+自营商城系统源码支持小程序+Android+IOS+H5
java·微信小程序·小程序·uni-app·音视频·交友
超浪的晨5 小时前
Java UDP 通信详解:从基础到实战,彻底掌握无连接网络编程
java·开发语言·后端·学习·个人开发
YouQian7725 小时前
Traffic Lights set的使用
算法