eolymp
bolt
Try our new interface for solving problems
Məsələlər

Уникальный цвет

Уникальный цвет

Дано дерево с $n$ вершинами, пронумерованными от $1$ до $n$. $i$ - ое ребро соединяет вершину $a_i$ и вершину $b_i$. Вершина $i$ окрашена в цвет $c_i$ (в этой задаче цвета представлены целыми числами). Вершина $x$ считается \textbf{хорошей}, если кратчайший путь от вершины $1$ до вершины $x$ не содержит вершину, окрашенную в тот же цвет, что и вершина $x$, кроме самой вершины $x$. Найдите все хорошие вершины. \InputFile Первая строка содержит количество вершин $n~(2 \le n \le 10^5)$. Вторая строка содержит цвета $c_1, c_2, ..., c_n~(1 \le c_i \le 10^5)$. Каждая из следующих $n - 1$ строк содержит два целых числа $a_i$ и $b_i~(1 \le a_i, b_i \le n)$. \OutputFile Выведите все хорошие вершины в виде целых чисел в порядке возрастания. Каждое число следует выводить в отдельной строке. \includegraphics{https://static.e-olymp.com/content/37/37efac8026a1c5f9a2c4c1892f30301006baf6b1.gif}
Zaman məhdudiyyəti 1 saniyə
Yaddaşı istafadə məhdudiyyəti 128 MiB
Giriş verilənləri #1
6
2 7 1 8 2 8
1 2
3 6
3 2
4 3
2 5
Çıxış verilənləri #1
1
2
3
4
6
Giriş verilənləri #2
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
Çıxış verilənləri #2
1
2
3
5
6
7
8