寶石的顏色 (AP325 P-8-7)

1.0s 256M

尋寶之旅的遊戲有一個地圖,地圖上有 \(n\) 個站,以 \(0\) 到 \(n-1\) 編號,此外有 \(n-1\) 條道路,這些道路都是單向的,遊戲固定從 \(0\) 號站出發,且已知從 \(0\) 號站出發可以直接或間接到達任何其他站。每一個站都有一顆寶石,寶石有分為多種顏色,第 \(i\) 站存放的寶石顏色為 \(c(i)\)。出發之前,你可以選定一種顏色的寶石收集箱,路途中遇到與你的收集箱相同顏色的寶石就可以收集(包含起點),請你計算最多可以收集到多少顆同色寶石。

輸入格式

第一行是一個正整數 \(n\),代表地圖上有 \(n\) 個站。第二行是 \(n\) 個非負整數,依序代表每一站的寶石顏色號碼 \(c(0), c(1), \ldots, c(n-1)\)。接著有 \(n-1\) 行,每行兩個以空白間隔的整數 \(s\) 與 \(t\),表示有一條 \(s\) 到 \(t\) 的道路。

限制

  • \(1 \le n \le 2 \times 10^5\)。
  • \(0 \le c(i) \le 10^9\)。
  • \(0 \le s, t \le n-1\),\(s \ne t\);保證從 \(0\) 號站可以到達每一站。

輸出格式

輸出為一整數,代表最多可能收集到的寶石數量。

評分說明

每筆計分測資獨立計分,每筆 5 分,共 100 分;範例不計分。

範例輸入 1

6
0 0 0 0 0 0
0 1
1 2
0 3
1 4
1 5

範例輸出 1

3

範例輸入 2

10
5 3 3 1 4 0 3 4 5 0
0 1
5 2
7 3
5 4
0 5
4 6
1 7
0 8
4 9

範例輸出 2

2

題目來源

本題出自中正大學吳邦一教授所著《AP325-從 APCS 實作題檢測三級到五級》(v1.5)第 8 章 P-8-7,第 292 頁;經作者同意於 AACPOJ 免費公開。教材下載:AP325 講義(Google Drive);Python 版:AP325-Python(HackMD)。原題出自 108 學年度全國高級中等學校資訊學科能力競賽(教材標註「108 全國高中賽」)。

題目頁說明

快速鍵

主要功能

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

限制

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