题目描述

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

示例

示例 1：

给定二叉树
          1
         / \
        2   3
       / \     
      4   5    
返回 3, 它的长度是路径 [4,2,1,3] 或者 [5,2,1,3]。
注意：两结点之间的路径长度是以它们之间边的数目表示。

解答方法

方法一：深度优先搜索

思路

这道题需要注意的一点是：二叉树的直径（即最长路径），不一定经过根结点。

二叉树的最长路径=max{左子树的最长路径,右子树的最长路径,左子树的深度+右子树的深度}

代码

class Solution:
    def diameterOfBinaryTree(self, root: TreeNode) -> int:
    
        if root is None:
            return 0
        res = self.depth(root.left) + self.depth(root.right)
        return max(self.diameterOfBinaryTree(root.left), self.diameterOfBinaryTree(root.right), res)
    def depth(self,root):
        if root is None:
            return 0
        return 1 + max(self.depth(root.left), self.depth(root.right))

时间复杂度

空间复杂度