标签：map val LCP preorder 44 二叉树 焰火 root

遍历+去重

package binarytree.numColor;

import binarytree.untils.TreeNode;

import java.util.HashMap;
import java.util.Map;

/**
 * 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
 */
public class numColor {
    /**
     * 遍历+去重
     * @param root
     * @return
     */
    public static int numColor(TreeNode root) {
        Map<Integer, Integer> map = new HashMap<>();
        preorder(root,map);
        return map.size();
    }

    private static void preorder(TreeNode root, Map<Integer, Integer> map) {
        if (root == null) {
            return;
        }
        if (map.containsKey(root.val)) {
            int count = map.get(root.val);
            map.put(root.val, ++count);
        } else {
            map.put(root.val, 1);
        }
        preorder(root.left, map);
        preorder(root.right, map);
    }

}

标签：map,val,LCP,preorder,44,二叉树,焰火,root
来源： https://www.cnblogs.com/xiaoshahai/p/15701609.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。