LCP 44. 开幕式焰火 #
题目 #
- 「力扣挑战赛」开幕式开始了,空中绽放了一颗二叉树形的巨型焰火。 给定一棵二叉树
root
代表焰火,节点值表示巨型焰火这一位置的颜色种类。请帮小扣计算巨型焰火有多少种不同的颜色。
思路 #
模拟 #
代码 #
模拟 #
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
private Set<Integer> set = new HashSet<>();
private void traverse(TreeNode root) {
if (root == null) return;
set.add(root.val);
traverse(root.left);
traverse(root.right);
}
public int numColor(TreeNode root) {
traverse(root);
return this.set.size();
}
}