#F. 树上路径

    Type: Default 1000ms 256MiB

树上路径

You cannot submit for this problem because the contest is ended. You can click "Open in Problem Set" to view this problem in normal mode.

有一棵 nn 个点的树,第 ii 条边连接 aia_ibib_i ,第 ii 个点颜色为 cic_i

对于每个 k=1,2,...,nk=1,2,..., n ,计算经过颜色 kk 的简单路径的数目。

路径 iijj 与 路径 jjii 视为相同的路径。

输入格式

第一行一个整数 nn

第二行 nn 个整数 c1,c2,,cnc_1, c_2, \dots , c_n

接下来 n1n-1 行,每行两个整数 ai,bia_i, b_i,表示一条边。

输出格式

nn 行,每行一个整数。

样例输入1

3
1 2 1
1 2
2 3

样例输出1

5
4
0

样例解释1

pi,jp_{i,j} 表示端点为 iijj 的简单路径

经过颜色1的简单路径有 5条:p1,1,p1,2,p1,3,p2,3,p3,3p_{1,1}, p_{1,2}, p_{1,3}, p_{2,3}, p_{3,3}

经过颜色2的简单路径有 4条:p1,2,p1,3,p2,2,p2,3p_{1,2}, p_{1,3}, p_{2,2}, p_{2,3}

没有经过颜色3的简单路径。

样例输入2

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

样例输出2

18
15
0
14
23
0
23
0

### 数据范围

1n21051ai,bi,cin1 \le n \le 2*10^5; 1\le a_i, b_i, c_i \le n

中大计算机 1

Not Attended
Status
Done
Rule
IOI
Problem
6
Start at
2025-6-23 10:30
End at
2025-6-23 12:30
Duration
2 hour(s)
Host
Partic.
1