| 题目链接 | 2385. 感染二叉树需要的总时间 |
|---|---|
| 思路 | 识别为“树的直径” |
| 题解链接 | 从两次遍历到一次遍历（Python/Java/C++/Go/JS/Rust） |
| 关键点 | 1. 返回值为(感染时间, 是否包含感染源) 2. 分为三种情况：本节点为感染源、左儿子（或右儿子）为感染源、该子树无感染源 |
| 时间复杂度 | \(O(n)\) |
| 空间复杂度 | \(O(n)\) |
代码实现：
class Solution:def amountOfTime(self, root: Optional[TreeNode], start: int) -> int:answer = 0def dfs(node):if node is None:return 0, Falsel_len, l_found = dfs(node.left)r_len, r_found = dfs(node.right)nonlocal answerif node.val == start:answer = max(l_len, r_len)return 1, Trueif l_found or r_found:answer = max(answer, l_len + r_len)return 1 + (l_len if l_found else r_len), Truereturn max(l_len, r_len) + 1, Falsedfs(root)return answer
