python-leetcode-二叉树的直径

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  # 返回最大直径