#CSES1139. 不同颜色
不同颜色
题目背景
翻译自 CSES-1139 题。
题目描述
给定一个包含 个节点的树,节点编号为 ,其中节点 是根节点。每个节点都有一个颜色。
你的任务是计算每个节点子树中不同颜色的数量。
输入格式
第一行包含一个整数 :树中的节点数量。节点编号为 ,根节点为节点 。
第二行包含 个整数 :每个节点的颜色。
接下来有 行,每行包含两个整数 和 :表示节点 和节点 之间有一条边。
输出格式
输出 个整数:对于每个节点 ,输出其子树中不同颜色的数量。
样例
5
2 3 2 2 1
1 2
1 3
3 4
3 5
3 1 2 1 1
说明/提示
;
;
。