题目
给你一棵二叉树的根节点,返回该树的 直径 。
二叉树的 直径 是指树中任意两个节点之间最长路径的 长度 。这条路径可能经过也可能不经过根节点 root 。
两节点之间路径的 长度 由它们之间边数表示。
代码
方法:递归
计算二叉树的直径可以理解为计算左、右子树的深度减一相加,然后再加上根节点左右后条边。
class Solution:def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:diameter = 0def dfs(node):nonlocal diameterif not node:return -1left_depth = dfs(node.left)right_depth = dfs(node.right)diameter = max(diameter, left_depth + right_depth + 2)return max(left_depth, right_depth) + 1dfs(root)return diameter