2385. 感染二叉树需要的总时间

合集 - LeetCode 树(11)

1.687. 最长同值路径2024-09-112.124. 二叉树中的最大路径和2024-09-11

3.2385. 感染二叉树需要的总时间2024-09-11

4.【模板题】- 144. 二叉树的前序遍历2024-09-125.【模板题】- 94. 二叉树的中序遍历2024-09-126.【模板题】- 145. 二叉树的后序遍历2024-09-127.872. 叶子相似的树2024-09-128.LCP 44. 开幕式焰火2024-09-129.404. 左叶子之和2024-09-1210.671. 二叉树中第二小的节点2024-09-1211.662. 二叉树最大宽度2024-09-24

收起

题目链接	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 = 0
        def dfs(node):
            if node is None:
                return 0, False
            l_len, l_found = dfs(node.left)
            r_len, r_found = dfs(node.right)
            nonlocal answer
            if node.val == start:
                answer = max(l_len, r_len)
                return 1, True
            if l_found or r_found:
                answer = max(answer, l_len + r_len)
                return 1 + (l_len if l_found else r_len), True
            return max(l_len, r_len) + 1, False
        dfs(root)
        return answer