【二叉树】开幕式焰火

0x00 题目

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

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

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

---

0x01 思路

统计不同元素的 个数
使用 集合 进行处理即可

---

0x02 解法

语言：Swift

树节点：TreeNode

public class TreeNode {
    public var val: Int
    public var left: TreeNode?
    public var right: TreeNode?
    public init() { self.val = 0; self.left = nil; self.right = nil; }
    public init(_ val: Int) { self.val = val; self.left = nil; self.right = nil; }
    public init(_ val: Int, _ left: TreeNode?, _ right: TreeNode?) {
        self.val = val
        self.left = left
        self.right = right
    }
}

解法：

func numColor(_ root: TreeNode?) -> Int {

    var arr = [Int]()

    func dfs(_ root: TreeNode?) {
        guard let root = root else { return }

        // 先放在数组中
        arr.append(root.val)
        
        // 递归遍历
        dfs(root.left)
        dfs(root.right)
    }

    dfs(root)
    
    // 最后一次性转成集合
    return Set(arr).count
}

---

小编辑器应用

---