原題頁面

CSES 3149 - Tree Coin Collecting II

1.0s 512M

給你一棵有 \(n\) 個節點的樹。某些節點上有一枚硬幣。

你的任務是回答 \(q\) 筆形式如下的詢問：從節點 \(a\) 到節點 \(b\)、而且會經過所有有硬幣的節點的路徑，最短的長度是多少？

輸入格式

第一行有兩個整數 \(n\) 與 \(q\)：節點數量與詢問數量。節點編號為 \(1, 2, \dots, n\)。

第二行有 \(n\) 個整數 \(c_1, c_2,\dots, c_n\)。若 \(c_i = 1\)，表示節點 \(i\) 上有硬幣；若 \(c_i = 0\)，表示節點 \(i\) 上沒有硬幣。你可以假設至少有一個節點上有硬幣。

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

最後有 \(q\) 行描述詢問。每行有兩個整數 \(a\) 與 \(b\)：起點與終點。

輸出格式

輸出 \(q\) 個整數：各筆詢問的答案。

範例輸入 1

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

範例輸出 1

6
5
6
8

限制

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

題目來源

CSES - Tree Coin Collecting II

題目來自 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，但同題短時間內多次提交會被視為刷分。