101635: [AtCoder]ABC163 F - path pass i

Memory Limit:256 MB Time Limit:2 S
Judge Style:Text Compare Creator:
Submit:0 Solved:0

Description

Score : $600$ points

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.


Sample Input 1

3
1 2 1
1 2
2 3

Sample 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.


Sample Input 2

1
1

Sample Output 2

1

Sample Input 3

2
1 2
1 2

Sample Output 3

2
2

Sample Input 4

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

Sample Output 4

5
8
10
5
5

Sample 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

Sample Output 5

18
15
0
14
23
0
23
0

Input

题意翻译

## 题目描述 给定一棵 $n$ 个点的树,给第 $i$ 个点染上颜色 $c_i$,其中,$c_i$ 为 $[1,n]$ 的一个整数。 现在,对于每一种颜色 $k$,你要求出有多少条简单路径满足路径上至少有一个点的颜色为 $k$。 ## 输入格式 第一行一个整数 $n$。 接下来一行 $n$ 个整数,表示 $c_i$。 接下来第 $3$ 到第 $n+1$ 行,每行两个整数 $u_i,v_i$,描述一条树边。 ## 输出格式 输出 $n$ 行,一行一个整数,分别表示对于颜色 $1,2,...,n$ 的答案。

加入题单

上一题 下一题 算法标签: