#CSES1139. 不同颜色

    ID: 293 传统题 1000ms 256MiB 尝试: 0 已通过: 0 难度: 3 上传者: 标签>树上问题DSU on tree树上启发式合并子树查询CSES

不同颜色

题目背景

翻译自 CSES-1139 题。

题目描述

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

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

输入格式

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

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

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

输出格式

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

样例

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

数据范围

  • 1n21051 \le n \le 2 \cdot 10^5
  • 1a,bn1 \le a,b \le n
  • 1ci1091 \le c_i\le 10^9