---
tags: CSES題解
---

# CSES Problem Set — Distinct Colors 題解

## 題目
給定一棵 $n$ 個節點的有根樹（以 $1$ 為根），每個節點被編號為 $1, 2, \ldots n$，並有一個顏色。

### 輸入
- 第一行有一個正整數 $n (1 \le n \le 2 \times 10^5)$
- 第二行有 $n$ 個正整數 $c_i$ 代表第 $i$ 個節點的顏色
- 最後有 $n-1$ 行，每行有兩個正整數 $u, v$，代表 $u$ 和 $v$ 相連

### 輸出
輸出 $n$ 個正整數，第 $i$ 個數字代表 $i$ 的子樹有多少種顏色

## 範例測資

Input:
5
2 3 2 2 1
1 2
1 3
3 4
3 5

Output:
3 1 2 1 1

## 觀察

## 想法
// 別忘了總結時間複雜度。

## 程式
`\``cpp=

`\``