CSES 2087 - Houses and Schools

1.0s 512M

一條街上有 \(n\) 棟房子,編號為 \(1,2,\dots,n\)。房子 \(a\) 和房子 \(b\) 的距離是 \(|a-b|\)。你已經知道每棟房子裡有幾個小孩。

你的任務是設立 \(k\) 所學校,每所學校都要設在某一棟房子裡。接著每個小孩都會走到離他最近的學校。在最佳的設法下,所有小孩走的總距離最少是多少?

輸入格式

第一行有兩個整數 \(n\) 和 \(k\):房子的數量與學校的數量。房子編號為 \(1,2\dots,n\)。

接下來有 \(n\) 個整數 \(c_1,c_2,\dots,c_n\):每棟房子裡的小孩人數。

輸出格式

輸出最小的總距離。

範例輸入 1

6 2
2 7 1 4 6 4

範例輸出 1

11

說明:學校會設在第 2 棟和第 5 棟房子。

限制

  • \(1 \le k \le n \le 3000\)
  • \(1 \le c_i \le 10^9\)

題目來源

CSES - Houses and Schools

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