LCP 44. 开幕式焰火

合集 - LeetCode 树(11)

1.687. 最长同值路径2024-09-112.124. 二叉树中的最大路径和2024-09-113.2385. 感染二叉树需要的总时间2024-09-114.【模板题】- 144. 二叉树的前序遍历2024-09-125.【模板题】- 94. 二叉树的中序遍历2024-09-126.【模板题】- 145. 二叉树的后序遍历2024-09-127.872. 叶子相似的树2024-09-12

8.LCP 44. 开幕式焰火2024-09-12

9.404. 左叶子之和2024-09-1210.671. 二叉树中第二小的节点2024-09-1211.662. 二叉树最大宽度2024-09-24

收起

题目链接	LCP 44. 开幕式焰火
思路	树的遍历
题解链接	Go DFS+哈希表
关键点	无
时间复杂度	$O(n)$
空间复杂度	$O(n)$

代码实现：

class Solution:
    def numColor(self, root: TreeNode) -> int:
        seen = set()
        def dfs(node):
            if node is None:
                return
            seen.add(node.val)
            dfs(node.left)
            dfs(node.right)
        dfs(root)
        return len(seen)