小猪佩奇爬树

题目描述

佩奇和乔治在爬♂树。 给定 $n$ 个节点的树 $T(V,E)$,第 $i$ 个节点的颜色为 $w_i$,保证有$1 \leq w_i \leq n$。 对于$1 \leq i \leq n$,分别输出有多少对点对 $(u,v)$,满足 $u<v$,且恰好经过**所有**颜色为 $i$ 的节点,对于节点颜色不为 $i$ 的其他节点,经过或不经过均可。 树上路径 $(u,v)$ 定义为序列 $\{f\}$,满足 $f_1=u,f_{|f|}=v$,且 $\forall 1 \leq i < |f|$,$T$ 中均存在边 $(f_i,f_{i+1})$,且 $\{f\}$ 中无重复元素,能够证明对于任意点对 $(u,v)$,其树上路径唯一。

输入输出格式

输入格式


第一行 $1$ 个正整数,表示 $n$ 。 第二行 $n$ 个正整数,第 $i$ 个正整数表示 $w_i$。 之后 $n-1$ 行,每行 $2$ 个正整数 $u,v$,表示 $T$ 中存在边 $(u,v)$。

输出格式


共 $n$ 行,每行 $1$ 个正整数,第 $i$ 行输出包含所有颜色为 $i$ 的节点的路径个数。

输入输出样例

输入样例 #1

4
1 2 2 3
1 2
2 3
3 4

输出样例 #1

3
4
3
6

输入样例 #2

10
9 7 4 2 3 4 4 5 8 5
2 1
3 2
4 2
5 2
6 4
7 4
8 1
9 4
10 4

输出样例 #2

45
35
9
0
1
45
34
9
17
45

说明

![](https://i.loli.net/2019/10/06/H9LuWl7GSXfs4M6.png) 对于第一组样例而言。 对于颜色 $1$,点对 $(1,2),(1,3),(1,4)$ 满足条件。 对于颜色 $2$,点对 $(1,3),(1,4),(2,3),(2,4)$ 满足条件。 对于颜色 $3$,点对 $(1,4),(2,4),(3,4)$ 满足条件。 对于颜色 $4$,由于图中没有颜色为 $4$ 的节点,所以所有点对均满足条件。 ### 数据范围 对于 $40\%$ 的数据, $n \leq 10^2$ 对于 $60\%$ 的数据, $n \leq 10^3$ 对于 $100\%$ 的数据, $n \leq 10^6$