题目地址(543. 二叉树的直径)

题目地址(543. 二叉树的直径)

https://leetcode.cn/problems/diameter-of-binary-tree/

题目描述

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

 

示例 :
给定二叉树

          1
         / \
        2   3
       / \     
      4   5    


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

 

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

前置知识

公司

• 暂无

思路

关键点

代码

• 语言支持：Python3

Python3 Code:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        """
        给定一颗二叉树，计算它的直径长度（任意两个结点路径长度中的最大值，路径可以穿过或者不穿过根结点）
        解决办法：
        1. 双重递归
        2. 对于某个结点，左子树的最大直径+右子树最长直径=该结点的最大直径
        """
        self.ans = 1
        def depth(node):
            if not node:
                return 0
            l = depth(node.left)
            r = depth(node.right)
            self.ans = max(self.ans,l+r+1)
            return max(l,r)+1
        depth(root)
        return self.ans-1
            

复杂度分析

令 n 为数组长度。

• 时间复杂度：$O(n)$
• 空间复杂度：$O(n)$