开幕式焰火

/*
 题目思路：
 使用Java Stream API来解决此问题。
 具体步骤如下：
 1. 将二叉树节点值收集到一个List中。
 2. 使用Stream去重并统计不同的颜色种类数。
 */
import java.util.*;
import java.util.stream.Collectors;

class TreeNode {
    int val;
    TreeNode left;
    TreeNode right;
    TreeNode(int x) { val = x; }
}

public class FireworkColorsStream {
    public int numOfColors(TreeNode root) {
        List<Integer> colors = new ArrayList<>();
        collectColors(root, colors);
        return colors.stream().collect(Collectors.toSet()).size();
    }

    private void collectColors(TreeNode node, List<Integer> colors) {
        if (node == null) return;
        colors.add(node.val);
        collectColors(node.left, colors);
        collectColors(node.right, colors);
    }

    public static void main(String[] args) {
        TreeNode root = new TreeNode(1);
        root.left = new TreeNode(3);
        root.right = new TreeNode(2);
        root.left.left = new TreeNode(1);
        root.right.left = new TreeNode(2);

        FireworkColorsStream solution = new FireworkColorsStream();
        System.out.println(solution.numOfColors(root)); // 输出: 3
    }
}