IT Кубок Кыргызстана 2022 Senior, Финал

Кыргызстан, 14 мая 2022

Задача E. Путь проходить i

Имя входного файла: стандартный ввод
Имя выходного файла: стандартный вывод
Ограничение по времени: 2 секунды
Ограничение по памяти: 256 мегабайт
У нас есть дерево с N вершинами, пронумерованными от 1 до N . i-е ребро в этом дереве соединяет
вершины ai и bi . Кроме того, каждая вершина окрашена в свой цвет, а цвет вершины i равен ci .
Здесь цвет каждой вершины представлен целым числом от 1 до N (включительно).
Для каждого k = 1, 2, ..., N решить следующую задачу:

• Найдите количество простых путей, которые проходят через вершину, окрашенную в k-й цвет,
один или несколько раз.

Формат входных данных

Первая срока содержит N количество вершин в дереве.1 6 N 6 2 ∗ 105 . Вторая строка содержит
n чисел c1 , . . . , cn (1 6 ci 6 N ). В следующих N − 1 строках заданы ai и bi . 1 6 ai , bi 6 N .
Формат выходных данных
Выведите ответы для k = 1, 2, ..., N по порядку, каждый в отдельной строке.
Примеры
стандартный ввод стандартный вывод
3 5
1 2 1 4
1 2 0
2 3
1 1
1
5 5
1 2 3 4 5 8
1 2 10
2 3 5
3 4 5
3 5

Замечание
Простые пути из вершины u в v и из v в u не различаются.

Страница 6 из 9