day15 二叉树的直径(leetcode)

matrix_studio

于 2021-12-23 20:43:31 发布

阅读量973

点赞数 1

CC 4.0 BY-SA版权

分类专栏： 2021-2022算法练习文章标签： leetcode 算法职场和发展

本文链接：https://blog.youkuaiyun.com/matrix_studio/article/details/122116111

2021-2022算法练习专栏收录该内容

31 篇文章

订阅专栏

本文介绍了一种通过递归算法来解决二叉树直径计算问题的方法。二叉树的直径是指从树中一个节点到另一个节点经过的最长路径。文章提供了一个具体的例子，并详细解释了如何使用递归函数计算该直径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

543. 二叉树的直径

题目：

给定一棵二叉树，你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过也可能不穿过根结点。

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意：两结点之间的路径长度是以它们之间边的数目表示。

解题：递归

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        def func(root):
        	# 叶节点终止递归，最大深度与最大直径置零
            if not root:
                return 0, 0
            # 获取当前最大直径和左右节点的最大深度
            left_deep, a = func(root.left)
            right_deep, b = func(root.right)
            # 更新最大直径
            max_len = max(left_deep+right_deep, a, b)
            # 更新最大深度
            max_deep = max(left_deep, right_deep) + 1
            # 返回上一层
            return max_deep, max_len
        # 返回结果
        return func(root)[1]