#abc163f. [abc163_f]path pass i

[abc163_f]path pass i

問題文

11 から NN までの番号がつけられた NN 個の頂点を持つ木があります。この木の ii 番目の辺は頂点 aia_ibib_i を結んでいます。 また、各頂点には色が塗られており、 頂点 ii に塗られている色は cic_i です。ここで、各頂点に塗られている色は 11 以上 NN 以下の整数で表されており、同じ整数は同じ色に、異なる整数は異なる色に対応します。

k=1,2,...,Nk=1,2,...,N に対して、以下の問題を解いてください。

  • kk が塗られている頂点を一度以上通るような単純パスの数を求めよ

補足: 頂点 uu から頂点 vv へ向かう単純パスと、頂点 vv から頂点 uu へ向かう単純パスは区別しません。

制約

  • 1leqNleq2times1051 \\leq N \\leq 2 \\times 10^5
  • 1leqcileqN1 \\leq c_i \\leq N
  • 1leqai,bileqN1 \\leq a_i,b_i \\leq N
  • 与えられるグラフは木である。
  • 入力はすべて整数である。

入力

入力は以下の形式で標準入力から与えられる。

NN c1c_1 c2c_2 ...... cNc_N a1a_1 b1b_1 :: aN1a_{N-1} bN1b_{N-1}

出力

k=1,2,...,Nk=1,2,...,N に対する問題の答えを、順番に改行区切りで出力せよ。


入力例 1

3
1 2 1
1 2
2 3

出力例 1

5
4
0

頂点 ii と頂点 jj を結ぶ単純パスを、Pi,jP_{i,j} と表します。

11 が塗られている頂点を一度以上通る単純パスは、
P1,1,,,P_{1,1}\\,,\\, P1,2,,,P_{1,2}\\,,\\, P1,3,,,P_{1,3}\\,,\\, P2,3,,,P_{2,3}\\,,\\, P3,3P_{3,3}
55 つあります。

22 が塗られている頂点を一度以上通る単純パスは、
P1,2,,,P_{1,2}\\,,\\, P1,3,,,P_{1,3}\\,,\\, P2,2,,,P_{2,2}\\,,\\, P2,3P_{2,3}
44 つあります。

33 が塗られている頂点を一度以上通る単純パスはありません。


入力例 2

1
1

出力例 2

1

入力例 3

2
1 2
1 2

出力例 3

2
2

入力例 4

5
1 2 3 4 5
1 2
2 3
3 4
3 5

出力例 4

5
8
10
5
5

入力例 5

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

出力例 5

18
15
0
14
23
0
23
0