题目:
给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。
示例 :
给定二叉树
1
/ \
2 3
/ \
4 5
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意:两结点之间的路径长度是以它们之间边的数目表示。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/diameter-of-binary-tree
思路:
可以将这个问题分成几个子问题
分别求左子树的直径,右子树的直径,本身的直径
所以最长的直径 = max(左子树直径,右子树直径,树本身直径)
树本身的直径 = 右子数深度 + 左子树深度
那这个题就变成了,用递归求左右子树直径和深度的题
(树深度 = max(左子树深度,右子树深度) +1)
上代码:
class Solution:
def diameterOfBinaryTree(self,root):
return self.aaa(root)[1]
def aaa(self,root):
if root is None:
return (0, 0)
#递归求子树深度,直径
left_dep, left_road = self.aaa(root.left)
right_dep, right_rodd = self.aaa(root.right)
#深度和直径的计算
depth = max(left_dep, right_dep) + 1
road = max(left_road, right_rodd, left_dep + right_dep)
return depth, road