链接: https://cses.fi/problemset/task/1139

板块: Tree Algorithms

时限: 1.00 s ｜ 内存: 512 MB

题目描述

给定一棵包含 $n$ 个节点的有根树。节点编号为 $1,2,\ldots,n$，且节点 $1$ 为根。每个节点有一种颜色。

你的任务是求出每个节点的子树中不同颜色的个数。

输入

第一行输入一个整数 $n$：节点个数。节点编号为 $1,2,\ldots,n$。

下一行有 $n$ 个整数 $c_1,c_2,\ldots,c_n$：每个节点的颜色。

接下来有 $n-1$ 行描述边。每行包含两个整数 $a$ 和 $b$：节点 $a$ 与 $b$ 之间有一条边。

输出

输出 $n$ 个整数：对于节点 $1,2,\ldots,n$，依次输出不同颜色的个数。

数据范围

$1 \le n \le 2 \cdot 10^5$ $1 \le a,b \le n$ $1 \le c_i \le 10^9$

样例输入

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

样例输出

3 1 2 1 1