Distinct Colors

题目描述

给定一棵有根树，包含 $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，输出其子树中不同颜色的数量。

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

3 1 2 1 1

提示

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

标签: CSES1139|树

来源

CSES1139|树