LeetCode-LCP 44. 开幕式焰火_Python

_trickle

已于 2022-06-07 13:07:26 修改

阅读量325

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签： leetcode 算法职场和发展

于 2022-06-07 13:07:06 首次发布

本文链接：https://blog.csdn.net/weixin_45344430/article/details/125163301

LeetCode 专栏收录该内容

348 篇文章

订阅专栏

该博客介绍了一个关于二叉树的问题，即给定一棵代表焰火的二叉树，计算其中不同颜色的种类数量。提供的解决方案通过前序遍历收集节点值并去除重复来得出答案。示例展示了如何处理不同情况，并给出了限制条件，如节点数量和值的范围。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

「力扣挑战赛」开幕式开始了，空中绽放了一颗二叉树形的巨型焰火。
给定一棵二叉树 root 代表焰火，节点值表示巨型焰火这一位置的颜色种类。请帮小扣计算巨型焰火有多少种不同的颜色。

示例 1：

输入：root = [1,3,2,1,null,2]
输出：3
解释：焰火中有 3 个不同的颜色，值分别为 1、2、3

示例 2：

输入：root = [3,3,3]
输出：1
解释：焰火中仅出现 1 个颜色，值为 3

提示：

1 <= 节点个数 <= 1000
1 <= Node.val <= 1000

程序代码

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
class Solution:
    def numColor(self, root: TreeNode) -> int:
        def preOrder(root, arr):
            if root is None:
                return
            arr.append(root.val)
            preOrder(root.left, arr)
            preOrder(root.right, arr)
        arr = []
        preOrder(root, arr)
        return len(set(arr))