Type: Default 2000ms 256MiB

不同颜色

You cannot submit for this problem because the contest is ended. You can click "Open in Problem Set" to view this problem in normal mode.

题目背景

翻译自 CSES-1139 题。

题目描述

给定一个包含 nn 个节点的树,节点编号为 1,2,,n1,2,…,n,其中节点 11 是根节点。每个节点都有一个颜色。

你的任务是计算每个节点子树中不同颜色的数量。

输入格式

第一行包含一个整数 nn:树中的节点数量。节点编号为 1,2,,n1,2,…,n,根节点为节点 11

第二行包含 nn 个整数 c1,c2,,cnc_1,c_2,…,c_n:每个节点的颜色。

接下来有 n1n−1 行,每行包含两个整数 aabb:表示节点 aa 和节点 bb 之间有一条边。

输出格式

输出 nn 个整数:对于每个节点 1,2,,n1,2,…,n,输出其子树中不同颜色的数量。

样例

5
2 3 2 2 1
1 2
1 3
3 4
3 5
3 1 2 1 1

说明/提示

1n21051 \leq n \leq 2 \cdot 10^5

1a,bn1 \leq a,b \leq n

1ci1091 \leq c_i\leq 10^9

树上启发式合并

Not Claimed
Status
Done
Problem
6
Open Since
2025-10-21 0:00
Deadline
2025-12-31 23:59
Extension
24 hour(s)