543. 二叉树的直径 - 力扣（LeetCode）
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:self.max_diameter = 0 # 记录最大直径def depth(node):if not node:return 0 # 叶子节点的深度为 0left_depth = depth(node.left) # 递归计算左子树深度right_depth = depth(node.right) # 递归计算右子树深度# 更新最大直径，直径等于左子树深度 + 右子树深度self.max_diameter = max(self.max_diameter, left_depth + right_depth)# 返回当前子树的深度return max(left_depth, right_depth) + 1depth(root) # 计算整棵树的深度return self.max_diameter # 返回最大直径
