題目連結 | 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