1142 : Tree Intersection
Time Limit: 2 Sec Memory Limit: 128 MB Submitted: 5 Solved: 2Description
Bobo has a tree with n vertices numbered by 1,2,…,n and (n-1) edges. The i-th vertex has color ci, and the i-th edge connects vertices ai and bi.
Let C(x,y) denotes the set of colors in subtree rooted at vertex x deleting edge (x,y).
Bobo would like to know Ri which is the size of intersection of C(ai, bi) and C(bi, ai) for all 1 ≤ i ≤ (n − 1). (i.e. |C(ai, bi) ∩ C(bi, ai)|)
Input
The input contains at most 15 sets. For each set:
The first line contains an integer n(2 ≤ n ≤ 105).
The second line contains n integers c1, c2, …, cn(1 ≤ ci ≤ n).
The i-th of the last (n-1) lines contains 2 integers ai, bi(1 ≤ ai, bi ≤ n).
Output
For each set, (n-1) integers R1, R2, …, Rn − 1.
Sample
4 1 2 2 1 1 2 2 3 3 4 5 1 1 2 1 2 1 3 2 3 3 5 4 5
1 2 1 1 1 2 1
Hint
Author
ftiasch