#P5588. 小猪佩奇爬树

    ID: 4552 Type: RemoteJudge 3000ms 500MiB Tried: 0 Accepted: 0 Difficulty: 5 Uploaded By: Tags>树形数据结构洛谷月赛

小猪佩奇爬树

题目描述

佩奇和乔治在爬♂树。

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

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

树上路径 (u,v)(u,v) 定义为序列 {f}\{f\},满足 f1=u,ff=vf_1=u,f_{|f|}=v,且 1i<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

之后 n1n-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\% 的数据, n102n \leq 10^2

对于 60%60\% 的数据, n103n \leq 10^3

对于 100%100\% 的数据, n106n \leq 10^6