atcoder#ABC198E. [ABC198E] Unique Color

[ABC198E] Unique Color

配点 : 500500

問題文

NN 頂点からなる木が与えられます。ii 番目の辺は頂点 AiA_i と頂点 BiB_i をつないでいます。頂点 ii は色 CiC_i で塗られています (この問題では、色は整数として表されます)。

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

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

制約

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

入力

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

NN

C1C_1 \ldots CNC_N

A1A_1 B1B_1

\vdots

AN1A_{N-1} BN1B_{N-1}

出力

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

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

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

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