day15 二叉树的直径(leetcode)

543. 二叉树的直径

题目：

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

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。

注意：两结点之间的路径长度是以它们之间边的数目表示。

解题：递归

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
        def func(root):
        	# 叶节点终止递归，最大深度与最大直径置零
            if not root:
                return 0, 0
            # 获取当前最大直径和左右节点的最大深度
            left_deep, a = func(root.left)
            right_deep, b = func(root.right)
            # 更新最大直径
            max_len = max(left_deep+right_deep, a, b)
            # 更新最大深度
            max_deep = max(left_deep, right_deep) + 1
            # 返回上一层
            return max_deep, max_len
        # 返回结果
        return func(root)[1]