|
发表于 2023-6-16 13:46:35
|
显示全部楼层
本楼为最佳答案
这种问题被称为求二叉树的直径,二叉树的直径被定义为树中任意两节点之间最长路径的长度。此问题可以通过深度优先搜索 (DFS) 的方法解决。
我们可以遍历每个节点,对于每个节点,我们计算左右子树的深度,然后相加。其中的最大值就是我们要找的最长路径。
以下是Python的代码实现:
- class TreeNode:
- def __init__(self, x):
- self.val = x
- self.left = None
- self.right = None
- class Solution:
- def __init__(self):
- self.max_diameter = 0
- def diameterOfBinaryTree(self, root: TreeNode) -> int:
- self.depth(root)
- return self.max_diameter
- def depth(self, node):
- if node is None:
- return 0
- # 分别计算左右子树的深度
- left_depth = self.depth(node.left)
- right_depth = self.depth(node.right)
- # 更新最大直径
- self.max_diameter = max(self.max_diameter, left_depth + right_depth)
- # 返回节点深度
- return max(left_depth, right_depth) + 1
复制代码
我们定义一个全局变量 max_diameter 来保存当前最大直径,然后通过递归的方式求出每个节点的左右子树的深度,这个深度就是从当前节点到其最远的叶子节点的最长路径。因为直径可能经过根节点,所以每次计算深度时,都要更新 max_diameter 。最后,递归遍历整个树, max_diameter 就是最长路径的长度。 |
|