news 2026/4/18 3:52:35

day167—递归—二叉树的直径(LeetCode-543)

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
day167—递归—二叉树的直径(LeetCode-543)

题目描述

给你一棵二叉树的根节点,返回该树的直径

二叉树的直径是指树中任意两个节点之间最长路径的长度。这条路径可能经过也可能不经过根节点root

两节点之间路径的长度由它们之间边数表示。

示例 1:

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

示例 2:

输入:root = [1,2]输出:1

提示:

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

解决方案:

这段代码是基于后序遍历(DFS)求解二叉树直径的经典实现,核心思路是通过递归计算每个节点的左右子树深度,同时统计以该节点为 “拐弯点” 的路径长度,最终得到整棵树的最大直径。

核心逻辑

  1. 核心定义

    • ans:全局变量,记录二叉树的最大直径(即所有路径中的最长长度);
    • dfs(node):递归函数,返回以node为根的子树的深度(从node到最远叶子节点的边数),同时在递归过程中更新全局最大直径。
  2. 递归边界:若node为空(!node),返回 0(空树深度为 0)。

  3. 后序遍历核心逻辑

    • 先递归计算左子树深度l_len = dfs(node->left)
    • 再递归计算右子树深度r_len = dfs(node->right)
    • 关键操作:以当前节点为 “拐弯点” 的路径长度是l_len + r_len(左子树深度 + 右子树深度),用该值更新全局最大值ans
    • 返回当前子树的深度:max(l_len, r_len) + 1(取左右子树深度的最大值,加 1 表示当前节点到子树的一条边)。
  4. 主函数逻辑:调用dfs(root)触发递归遍历整棵树,最终返回全局最大值ans(即二叉树的直径)。

关键特点

  • 时间复杂度 O (n):每个节点仅被递归访问一次,n 为节点数;
  • 空间复杂度 O (h):h 为树的高度,递归栈深度等于树高(平衡树 h=logn,退化为链表 h=n);
  • 逻辑简洁:将 “计算子树深度” 和 “统计最大直径” 融合在一次 DFS 中,无需额外遍历;
  • 直径定义:二叉树的直径是 “任意两个节点之间的最长路径长度”(路径长度 = 边数),该代码精准统计了所有可能的路径(以每个节点为拐弯点)。

验证示例(简单二叉树)

假设有如下二叉树:

plaintext

1 / \ 2 3 / \ 4 5
  • 递归到节点 2 时,l_len=1(节点 4)、r_len=1(节点 5),ans更新为1+1=2
  • 递归到节点 1 时,l_len=2(节点 2 的子树深度)、r_len=1(节点 3 的子树深度),ans仍为 2(2+1=3?此处修正:该例子中节点 1 的l_len=2r_len=1ans会更新为 3,对应路径 4-2-5(长度 2)或 4-2-1-3(长度 3),最终直径为 3);
  • 最终返回ans=3,符合实际最长路径长度。

总结

  1. 核心思路:通过后序遍历递归计算子树深度,同时统计每个节点作为 “拐弯点” 的路径长度,取最大值即为二叉树直径;
  2. 关键设计:dfs函数兼具 “返回子树深度” 和 “更新最大直径” 两个职责,是该问题的最优解法;
  3. 功能效果:能正确计算任意二叉树的直径,时间 / 空间效率均为最优。

函数源码:

/** * 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 ans=0; int dfs(TreeNode* node){ if(!node) return 0; int l_len=dfs(node->left); int r_len=dfs(node->right); ans=max(ans,l_len+r_len); return max(l_len,r_len)+1; } int diameterOfBinaryTree(TreeNode* root) { dfs(root); return ans; } };
版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/4/14 23:49:43

基于大数据的化妆品销售系统-计算机毕业设计源码+LW文档

基于大数据的化妆品销售系统 摘要&#xff1a;本文围绕基于大数据的化妆品销售系统展开论述&#xff0c;阐述了其研究背景意义、需求分析及功能设计。随着化妆品市场发展和大数据技术兴起&#xff0c;该系统能解决传统销售模式的问题&#xff0c;满足多方需求&#xff0c;通过大…

作者头像 李华
网站建设 2026/4/17 18:20:02

### 技术文章大纲:C语言造轮子大赛

背景与意义 C语言造轮子大赛旨在鼓励开发者深入理解底层原理&#xff0c;通过手动实现常见库或工具&#xff08;如字符串处理、数据结构、内存管理等&#xff09;提升编程能力。这类比赛通常考察代码效率、可读性、创新性及对标准库的替代价值。 常见轮子实现方向 基础数据结构…

作者头像 李华
网站建设 2026/4/18 0:49:01

Node.js用require.resolve优化模块加载

&#x1f493; 博客主页&#xff1a;瑕疵的CSDN主页 &#x1f4dd; Gitee主页&#xff1a;瑕疵的gitee主页 ⏩ 文章专栏&#xff1a;《热点资讯》 解锁Node.js模块加载效率&#xff1a;require.resolve的深度优化实践 目录 解锁Node.js模块加载效率&#xff1a;require.resolve…

作者头像 李华
网站建设 2026/4/8 13:52:03

pythonpython付费选座自习室小程序

目录付费选座自习室小程序的功能需求技术实现方案核心功能模块用户体验优化数据安全与性能商业化运营项目技术支持可定制开发之功能亮点源码获取详细视频演示 &#xff1a;文章底部获取博主联系方式&#xff01;同行可合作付费选座自习室小程序的功能需求 付费选座自习室小程序…

作者头像 李华
网站建设 2026/3/11 21:39:45

论文开题“黑科技”:书匠策AI如何让你的开题报告“一键起飞”?

写论文开题报告时&#xff0c;你是否也经历过这些“崩溃时刻”&#xff1f;对着空白文档发呆两小时&#xff0c;选题方向像无头苍蝇般乱撞&#xff1b;文献综述翻遍全网&#xff0c;关键信息却像散落的拼图难以整合&#xff1b;研究框架反复修改&#xff0c;逻辑链条始终不够清…

作者头像 李华