原題頁面

CSES 1139 - Distinct Colors

1.0s 512M

給你一棵由 \(n\) 個節點組成的有根樹。節點編號為 \(1, 2, \ldots, n\)，其中節點 \(1\) 是根。每個節點都有一個顏色。

你的任務是對於每個節點，求出該節點的子樹中相異顏色的數量。

輸入格式

第一行有一個整數 \(n\)：節點數量。節點編號為 \(1, 2, \ldots, n\)。

下一行有 \(n\) 個整數 \(c_1, c_2, \ldots, c_n\)：每個節點的顏色。

接下來有 \(n - 1\) 行描述邊。每行有兩個整數 \(a\) 和 \(b\)：節點 \(a\) 與節點 \(b\) 之間有一條邊。

輸出格式

輸出 \(n\) 個整數：對於每個節點 \(1, 2, \ldots, n\)，該節點子樹中相異顏色的數量。

範例輸入 1

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

範例輸出 1

3 1 2 1 1

限制

  • \(1 \le n \le 2 \cdot 10^5\)
  • \(1 \le a, b \le n\)
  • \(1 \le c_i \le 10^9\)

題目來源

題目來自 CSES Problem Set（Antti Laaksonen），授權 CC BY-NC-SA 4.0；本頁為翻譯，以相同授權分享。

題目頁說明

快速鍵

主要功能

  • 範例測試 — 執行題目附帶的範例測資並自動比對預期輸出。
  • 自訂測試 — 自己貼 stdin 執行程式。可勾選「與預期輸出比對 (diff)」做行對行比對。
  • 模板 — 貼上你在個人資料設定的預設程式碼模板。
  • 協作 — 與其他同學共筆編輯這題的程式碼。
  • 自動草稿 — 編輯器內容每 1.5 秒自動存到瀏覽器（per 帳號 / 題目 / 語言）。
  • 提交 — 把程式碼交給 judge 評測，回傳 AC / WA / TLE 等結果。

限制

  • 程式碼最多 65,536 字元
  • 自訂測試 stdin 與預期輸出各最多 1 MB （約 100 萬字元）
  • 自訂測試與範例測試共用一個沙箱，每人約 3 秒 1 次 （範例測試 1 秒 1 次）
  • 自訂測試與範例測試都有 15 秒 牆鐘上限（正式評測仍依題目原本時限）
  • 互動題不提供自訂測試（無法模擬與 judge 互動）。
  • 提交評測本身沒有 rate limit，但同題短時間內多次提交會被視為刷分。