#CSES1139. 不同颜色

不同颜色

题目背景

翻译自 CSES-1139 题。

题目描述

给定一个包含 nn 个节点的树,节点编号为 1,2,,n1,2,…,n,其中节点 11 是根节点。每个节点都有一个颜色。

你的任务是计算每个节点子树中不同颜色的数量。

输入格式

第一行包含一个整数 nn:树中的节点数量。节点编号为 1,2,,n1,2,…,n,根节点为节点 11

第二行包含 nn 个整数 c1,c2,,cnc_1,c_2,…,c_n:每个节点的颜色。

接下来有 n1n−1 行,每行包含两个整数 aabb:表示节点 aa 和节点 bb 之间有一条边。

输出格式

输出 nn 个整数:对于每个节点 1,2,,n1,2,…,n,输出其子树中不同颜色的数量。

样例

5
2 3 2 2 1
1 2
1 3
3 4
3 5
3 1 2 1 1

说明/提示

1n21051 \leq n \leq 2 \cdot 10^5

1a,bn1 \leq a,b \leq n

1ci1091 \leq c_i\leq 10^9