AtCoder Beginner Contest 163
翻译中…
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
题目信息
题面与样例版权归 AtCoder 及出题人所有;中文翻译由 AI 生成(CI 管线),如有出入请以原文为准。