【模板題】 543. 二叉樹的直徑

WrRan發表於2024-09-11
題目連結 543. 二叉樹的直徑
思路 經典模板題
題解連結 【影片】徹底掌握直徑 DP!從二叉樹到一般樹!(Python/Java/C++/Go)
關鍵點 1. 空節點返回\(-1\) 2. 本節點回收子節點結果時需要\(+1\) 3. 返回給父節點時是取\(\max(v_{l}, v_{r})\)
時間複雜度 \(O(n)\)
空間複雜度 \(O(n)\)

程式碼實現:

class Solution:
    def diameterOfBinaryTree(self, root: Optional[TreeNode]) -> int:
        answer = 0
        def dfs(node):
            if node is None:
                return -1
            l_len = dfs(node.left) + 1
            r_len = dfs(node.right) + 1
            nonlocal answer
            answer = max(answer, l_len+r_len)
            return max(l_len, r_len)
        dfs(root)
        return answer

相關文章