LCP 44-开幕式焰火

「力扣挑战赛」开幕式开始了，空中绽放了一颗二叉树形的巨型焰火。 给定一棵二叉树 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

Problem: LCP 44. 开幕式焰火

[TOC]

思路

• 这题本质上是让你统计，烟火有多少种不同的颜色，即这颗二叉树有多少个不同的值。
• 遍历二叉树，用set容器记录每个节点值的存在性，到最后直接取set.size()即可。

优化

以上思路尚可，但结果不尽人意，尤其是空间占用过高。不论是基于红黑树的set，还是基于哈希表的unordered_set，在这一题都有点“大材小用”，已知节点值在区间[1, 1000]之间的情况下，可以考虑直接创建1000个int的数组代替之。

Code

[]

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
    int array[1000] = {0};
public:
    int numColor(TreeNode* root) {
        order(root);    //树的遍历
        return accumulate(array, array+1000, 0);    //C++提供的累加器
    }

    //先序遍历，其实随便怎么遍历都行
    void order(TreeNode *root) {
        if (root) {
            array[root->val -1] = 1;    //有这一种颜色，记录之
            order(root->left);
            order(root->right);
        }
    }
};