n вершин с номерами от 0 до n - 1, корень — вершина 0. Список edges содержит n - 1 ребро: пара [a, b] значит, что вершины a и b соединены. Вершина i покрашена в цвет colors[i].
Поддерево вершины v — это сама v и все её потомки. Найдите вершину, у которой всё поддерево окрашено в один цвет, и верните наибольший возможный размер такого поддерева (число вершин в нём).edges=[[0, 1], [1, 2], [1, 3], [0, 4]], colors=[2, 5, 5, 5, 2] → 3
Поддерево вершины 1 — это вершины 1, 2, 3, все цвета 5: размер 3. У корня дети разных цветов, так что всё дерево не подходит.
edges=[[1, 0], [2, 0]], colors=[4, 4, 4] → 3
Всё дерево одного цвета — ответ 3. Рёбра можно записывать в любом направлении.
edges=[[0, 1], [1, 2]], colors=[1, 2, 3] → 1
Одноцветный только лист 2 — его поддерево из одной вершины.
edges[i] = [a, b], 0 ≤ a, b < nedges=[[0, 1], [1, 2], [1, 3], [0, 4]], colors=[2, 5, 5, 5, 2]
3
«Запустить» — первые 3 теста у тебя в браузере, «Отправить» — все 66 тестов на сервере, включая скрытые.