atcoder#ABC163F. [ABC163F] path pass i
[ABC163F] path pass i
Score : points
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.
3
1 2 1
1 2
2 3
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.
1
1
1
2
1 2
1 2
2
2
5
1 2 3 4 5
1 2
2 3
3 4
3 5
5
8
10
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
18
15
0
14
23
0
23
0