5297. 不同颜色（Distinct Colors）
时间限制：1000 MS 内存限制：256 MB
题目描述
## 题目描述 给定一棵包含 $n $个节点的根树。节点编号为 $1,2,\dots,n$，其中节点 $1 $是根。每个节点都有一个颜色。 你的任务是为每个节点计算**其子树中的不同颜色**数量。 ## 输入格式 第一行输入一个整数 $n$：节点数。节点编号为 $1,2,\dots,n$。 第二行包含$ n$ 个整数 $c_1,c_2,\dots,c_n$：每个节点的颜色。 接下来有$ n-1$ 行描述边。每行包含两个整数$ a$ 和 $b$：表示节点 $a $和 $b $之间有一条边。 ## 输出格式 输出 $n$ 个整数：对每个节点 $1,2,\dots,n$，输出其子树中的不同颜色数量。 ## 输入输出样例 ### 输入 #1 ``` 5 2 3 2 2 1 1 2 1 3 3 4 3 5 ``` ### 输出 #1 ``` 3 1 2 1 1 ``` ## 说明/提示 ### 数据规模与约定 - $1\len\le2\cdot10^5$ - $1\lea,b\len$ - $1\leci\le10^9$