【模板题】 543. 二叉树的直径
题目链接 | 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