#abc163f. [abc163_f]path pass i
[abc163_f]path pass i
Problem Statement
We have a tree with vertices numbered to . The -th edge in this tree connects Vertex and . Additionally, each vertex is painted in a color, and the color of Vertex is . Here, the color of each vertex is represented by an integer between and (inclusive). The same integer corresponds to the same color; different integers correspond to different colors.
For each , solve the following problem:
- Find the number of simple paths that visit a vertex painted in the color one or more times.
Note: The simple paths from Vertex to and from to are not distinguished.
Constraints
- The given graph is a tree.
- All values in input are integers.
Input
Input is given from Standard Input in the following format:
Output
Print the answers for 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 denote the simple path connecting Vertex and .
There are simple paths that visit a vertex painted in the color one or more times:
There are simple paths that visit a vertex painted in the color one or more times:
There are no simple paths that visit a vertex painted in the color 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