AtCoder Beginner Contest 163
F
path pass i
AtCoder
⌀ 2470
原题
← E→
翻译中…

Problem Statement

翻译中…

We have a tree with $N$ vertices numbered $1$ to $N$. The $i$-th edge in this tree connects Vertex $a_i$ and $b_i$. Additionally, each vertex is painted in a color, and the color of Vertex $i$ is $c_i$. Here, the color of each vertex is represented by an integer between $1$ and $N$ (inclusive). The same integer corresponds to the same color; different integers correspond to different colors.

翻译中…

For each $k=1, 2, ..., N$, solve the following problem:

翻译中…
  • Find the number of simple paths that visit a vertex painted in the color $k$ one or more times.
翻译中…

Note: The simple paths from Vertex $u$ to $v$ and from $v$ to $u$ are not distinguished.

翻译中…

Constraints

翻译中…
  • $1 \leq N \leq 2 \times 10^5$
  • $1 \leq c_i \leq N$
  • $1 \leq a_i,b_i \leq N$
  • The given graph is a tree.
  • All values in input are integers.
翻译中…

Input

翻译中…

Input is given from Standard Input in the following format:

翻译中…

N
c_1 c_2 ... c_N
a_1 b_1
:
a_{N-1} b_{N-1}

翻译中…

Output

翻译中…

Print the answers for $k = 1, 2, ..., N$ in order, each in its own line.

翻译中…

样例

Input #1
3
1 2 1
1 2
2 3
Output #1
5
4
0
Let $P_{i,j}$ denote the simple path connecting Vertex $i$ and $j$. There are $5$ simple paths that visit a vertex painted in the color $1$ one or more times: $P_{1,1}\,,\,$ $P_{1,2}\,,\,$ $P_{1,3}\,,\,$ $P_{2,3}\,,\,$ $P_{3,3}$ There are $4$ simple paths that visit a vertex painted in the color $2$ one or more times: $P_{1,2}\,,\,$ $P_{1,3}\,,\,$ $P_{2,2}\,,\,$ $P_{2,3}$ There are no simple paths that visit a vertex painted in the color $3$ one or more times.
Input #2
1
1
Output #2
1
Input #3
2
1 2
1 2
Output #3
2
2
Input #4
5
1 2 3 4 5
1 2
2 3
3 4
3 5
Output #4
5
8
10
5
5
Input #5
8
2 7 2 5 4 1 7 5
3 1
1 2
2 7
4 5
5 6
6 8
7 8
Output #5
18
15
0
14
23
0
23
0
题目信息
题号
abc163_f
时间限制
2 s
内存限制
1024 MB

题面与样例版权归 AtCoder 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。