atcoder#ABC198E. [ABC198E] Unique Color

[ABC198E] Unique Color

题目描述

N N 頂点からなる木が与えられます。i i 番目の辺は頂点 Ai A_i と頂点 Bi B_i をつないでいます。頂点 i i は色 Ci C_i で塗られています (この問題では、色は整数として表されます)。

頂点 1 1 から頂点 x x への最短パスに、頂点 x x と同じ色で塗られた頂点が頂点 x x 以外に存在しないとき、頂点 x x よい頂点 であるといいます。

よい頂点を全て求めてください。

输入格式

入力は以下の形式で標準入力から与えられる。

N N C1 C_1 \ldots CN C_N A1 A_1 B1 B_1 \vdots AN1 A_{N-1} BN1 B_{N-1}

输出格式

全てのよい頂点の番号を、昇順に改行区切りで出力せよ。

6
2 7 1 8 2 8
1 2
3 6
3 2
4 3
2 5
1
2
3
4
6
10
3 1 4 1 5 9 2 6 5 3
1 2
2 3
3 4
4 5
5 6
6 7
7 8
8 9
9 10
1
2
3
5
6
7
8

提示

制約

  • 2  N  105 2\ \leq\ N\ \leq\ 10^5
  • 1  Ci  105 1\ \leq\ C_i\ \leq\ 10^5
  • 1  Ai,Bi  N 1\ \leq\ A_i,B_i\ \leq\ N
  • 与えられるグラフは木である
  • 入力は全て整数

Sample Explanation 1

例えば頂点 1 1 から頂点 6 6 への最短パスには頂点 1,2,3,6 1,2,3,6 が含まれます。これらの中に、頂点 6 6 と同じ色の頂点は頂点 6 6 以外存在しないので、頂点 6 6 はよい頂点です。 一方で、頂点 1 1 から頂点 5 5 への最短パスには頂点 1, 2, 5 1,\ 2,\ 5 が含まれ、頂点 1 1 と頂点 5 5 の色は同じであるため、頂点 5 5 はよい頂点ではありません。