#P5588. 小猪佩奇爬树

小猪佩奇爬树

题目描述

佩奇和乔治在爬♂树。

给定 nn 个节点的树 T(V,E)T(V,E),第 ii 个节点的颜色为 wiw_i,保证有 1≤wi≤n1 \leq w_i \leq n。

对于 1≤i≤n1 \leq i \leq n,分别输出有多少对点对 (u,v)(u,v),满足 u<vu<v,且恰好经过所有颜色为 ii 的节点,对于节点颜色不为 ii 的其他节点,经过或不经过均可。

树上路径 (u,v)(u,v) 定义为序列 {f}\{f\},满足 f1=u,f∣f∣=vf_1=u,f_{|f|}=v,且 ∀1≤i<∣f∣\forall 1 \leq i < |f|,TT 中均存在边 (fi,fi+1)(f_i,f_{i+1}),且 {f}\{f\} 中无重复元素,能够证明对于任意点对 (u,v)(u,v),其树上路径唯一。

输入格式

第一行 11 个正整数,表示 nn 。

第二行 nn 个正整数,第 ii 个正整数表示 wiw_i。

之后 n−1n-1 行,每行 22 个正整数 u,vu,v,表示 TT 中存在边 (u,v)(u,v)。

输出格式

共 nn 行,每行 11 个正整数,第 ii 行输出包含所有颜色为 ii 的节点的路径个数。

4
1 2 2 3
1 2
2 3
3 4
3
4
3
6
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
45
35
9
0
1
45
34
9
17
45

提示

对于第一组样例而言。

对于颜色 11,点对 (1,2),(1,3),(1,4)(1,2),(1,3),(1,4) 满足条件。

对于颜色 22,点对 (1,3),(1,4),(2,3),(2,4)(1,3),(1,4),(2,3),(2,4) 满足条件。

对于颜色 33,点对 (1,4),(2,4),(3,4)(1,4),(2,4),(3,4) 满足条件。

对于颜色 44,由于图中没有颜色为 44 的节点,所以所有点对均满足条件。

数据范围

对于 40%40\% 的数据,n≤102n \leq 10^2。

对于 60%60\% 的数据,n≤103n \leq 10^3。

对于 100%100\% 的数据,n≤106n \leq 10^6。