力扣HOT100(45) 二叉树的直径

任意一条路径,都可以看作是以某个节点为 "拐点",左边接左子树的最深路径,右边接右子树的最深路径。

比如路径 4→2→1→3 的拐点是节点 1:

  • 左子树最深路径:1→2→4(2 条边)
  • 右子树最深路径:1→3(1 条边)
  • 总路径长度:2+1=3(边数)

所以我们的目标转化为:

  1. 遍历树中所有节点,计算每个节点作为 "拐点" 时的路径长度;
  2. 取所有路径长度的最大值,就是整棵树的直径

完整解题步骤

1. 定义两个核心变量

  • depth(node):递归函数,返回以node为根的子树的深度(节点数)
  • ans:全局变量,记录所有节点作为拐点时,路径经过的节点数最大值

2. 递归函数 depth(node) 的逻辑

  1. 终止条件 :如果node是空节点,返回 0(空树深度为 0)
  2. 递归左子树 :得到左子树深度L
  3. 递归右子树 :得到右子树深度R
  4. 更新最大值 :当前节点作为拐点的路径节点数 = L + R + 1(+1 是加上当前节点本身),用这个值更新ans
  5. 返回当前子树深度max(L, R) + 1(当前节点 + 左右子树中更深的那个)

3. 主函数逻辑

  1. 初始化ans = 1(至少有一个节点,路径节点数为 1)
  2. 调用depth(root)遍历整棵树
  3. 最终直径 = ans - 1(因为边数 = 节点数 - 1)
cpp 复制代码
class Solution {
    int ans; // 全局变量:记录所有路径的节点数最大值

    // 递归函数:返回以rt为根的子树的深度(节点数)
    int depth(TreeNode* rt){
        // 1. 终止条件:空节点,深度为0
        if (rt == NULL) {
            return 0;
        }

        // 2. 先算左子树深度
        int L = depth(rt->left); 
        // 3. 再算右子树深度
        int R = depth(rt->right); 

        // 4. 关键:计算当前节点作为拐点的路径节点数,更新最大值
        ans = max(ans, L + R + 1); 

        // 5. 返回当前子树的深度:左右子树中更深的那个 + 当前节点
        return max(L, R) + 1; 
    }

public:
    int diameterOfBinaryTree(TreeNode* root) {
        ans = 1; // 初始化为1,对应只有一个节点的情况
        depth(root); // 遍历整棵树
        return ans - 1; // 节点数最大值减1,得到边数(直径)
    }
};
相关推荐
geovindu4 分钟前
go: Floyd-Warshall Algorithms
开发语言·后端·算法·golang
CoderYanger9 分钟前
视频裁剪+缩放+自动添加水印脚本(Python版)
开发语言·后端·python·程序人生·职场和发展·音视频·学习方法
db_murphy12 分钟前
机器学习决策树的基尼系数是个啥?
学习·算法
拳里剑气19 分钟前
C++算法:优先级队列
开发语言·c++·算法·优先级队列
Turbo正则30 分钟前
机器学习入门笔记 | 基础算法及其应用场景
笔记·算法·机器学习
东华万里32 分钟前
第37篇 手撕二叉树与堆的底层逻辑
数据结构·算法
知无不研34 分钟前
算法:判断一个数是不是2的幂
数据结构·算法
luj_17689 小时前
残熵算法实时化三大瓶颈突破
c语言·开发语言·网络·经验分享·算法
mCell9 小时前
你以为短链接只是 Hash + 301/302?
后端·算法·架构
KaMeidebaby11 小时前
卡梅德生物技术快报|抗体亲和力成熟工业化调控新机制:差异性浆细胞增殖工艺优化思路
java·开发语言·人工智能·算法·机器学习·架构·spark