二叉树的直径（LeetCode）

好好学习Py

于 2024-09-01 23:57:36 发布

阅读量178

点赞数 9

分类专栏：算法与数据结构文章标签： leetcode 算法 python 数据结构

本文链接：https://blog.csdn.net/weixin_74254879/article/details/141792428

版权

算法与数据结构专栏收录该内容

80 篇文章 0 订阅

订阅专栏

题目

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

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

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

解题

class TreeNode:
    def __init__(self, val=0, left=None, right=None):
        self.val = val
        self.left = left
        self.right = right


def diameterOfBinaryTree(root):
    diameter = 0

    def depth(node):
        nonlocal diameter
        if not node:
            return 0

        left_depth = depth(node.left)
        right_depth = depth(node.right)

        # 更新直径
        diameter = max(diameter, left_depth + right_depth)

        # 返回节点的最大深度
        return max(left_depth, right_depth) + 1

    depth(root)
    return diameter


# 示例用法
root = TreeNode(1)
root.left = TreeNode(2)
root.right = TreeNode(3)
root.left.left = TreeNode(4)
root.left.right = TreeNode(5)

# 调用函数，输出结果
print(diameterOfBinaryTree(root))  # 输出3