【递归算法力扣python】---543.二叉树的直径

题目:

给定一棵二叉树,你需要计算它的直径长度。一棵二叉树的直径长度是任意两个结点路径长度中的最大值。这条路径可能穿过根结点。

示例 :
给定二叉树

      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

你可能感兴趣的:(算法)