力扣热题100之二叉树的直径

最新推荐文章于 2025-08-22 19:21:44 发布

Mi Manchi26

最新推荐文章于 2025-08-22 19:21:44 发布

阅读量294

点赞数 3

CC 4.0 BY-SA版权

文章标签： leetcode 算法 python

本文链接：https://blog.csdn.net/weixin_64233807/article/details/148429429

题目

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

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

两节点之间路径的长度由它们之间边数表示。
在这里插入图片描述

代码

方法：递归

计算二叉树的直径可以理解为计算左、右子树的深度减一相加，然后再加上根节点左右后条边。

class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        diameter = 0
        
        def dfs(node):
            nonlocal diameter
            if not node:
                return -1
            left_depth = dfs(node.left)
            right_depth = dfs(node.right)
            diameter = max(diameter, left_depth + right_depth + 2)
            return max(left_depth, right_depth) + 1
        
        dfs(root)
        return diameter