HackDream Purple 01-C: Tự tại

Xem PDF

Nộp bài

Điểm: 100 (thành phần)
Thời gian: 0.5s
Bộ nhớ: 256M
Input: bàn phím
Output: màn hình

Tác giả:
Dạng bài

Rotund có ~n~ khoảnh vườn cách đều trên một hàng ngang, đánh số từ ~1~ tới ~n~ từ trái qua phải. Vì sức một người có hạn, hiện tại Rotund mới trồng cây trên ~k~ khoảnh vườn (biểu diễn bởi số ~1~) và ~n-k~ khoảnh vườn còn lại vẫn đang để không (số khoảnh vườn đã trồng không vượt quá một nửa tổng số khoảnh vườn).

Sau một thời gian trồng cây, khu vực đất của ~k~ khoảnh cũ có dấu hiệu xói mòn dinh dưỡng và xuất hiện các loại sâu bệnh, Rotund quyết định di chuyển cây từ ~k~ khoảnh này sang ~k~ khoảnh vườn khác (chưa được trồng cây trước đó) để xử lý xong các khu đất này. Cứ mỗi khi di chuyển cây ở khoảnh vườn ~i~ sang khoảnh ~j~ thì Rotund cần trả cho đội vận chuyển chi phí là ~|j-i|~. Rotund muốn chi phí vận chuyển cần phải là nhỏ nhất có thể.

Yêu cầu

Cho ~n~ khoảnh vườn và số lượng khoảnh vườn đã được trồng cây ~k~. Tính chi phí vận chuyển tối thiểu để di dời toàn bộ cây sang ~k~ khoảnh vườn trống mới.

Input

  • Dòng đầu tiên chứa 2 số nguyên dương ~n~, ~k~ ~(1≤n≤10^6, 1≤k≤n/2)~ cách nhau một dấu cách.
  • Dòng thứ hai chứa ~n~ số nguyên ~a_i~ ~(0≤a_i≤1)~ thể hiện trạng thái của mỗi khoảnh vườn, cách nhau một dấu cách. ~0~ là khoảnh vườn hiện tại trống, ~1~ là khoảnh vườn hiện tại đang được trồng cây.

Output

Một dòng duy nhất chứa một số nguyên dương là chi phí ít nhất cần thiết để di chuyển toàn bộ cây trồng.

Sample Input

6 3
0 1 0 0 1 1

Sample Output

5

Giải thích

Có tổng cộng ~6~ khoảnh vườn và ~3~ khoảnh ở vị trí ~2~, ~5~, ~6~ đã có cây trồng.

Chuyển cây ở khoảnh ~2~ sang khoảnh ~1~ mất chi phí ~1~.

Chuyển cây ở khoảnh ~5~ sang khoảnh ~3~ mất chi phí ~2~.

Chuyển cây ở khoảnh ~6~ sang khoảnh ~4~ mất chi phí ~2~.

Tổng chi phí cần là ~5~.

Subtask

  • Có 30% số test ứng với 30% số điểm có ~1≤n≤20~;
  • Có 30% số test ứng với 30% số điểm có ~1≤n≤10^2~;
  • Có 20% số test ứng với 20% số điểm có ~1≤n≤10^3~;
  • 20% số test còn lại tương ứng với 20% số điểm không có giới hạn gì thêm.

Bình luận đầu tiên

Bình luận

Không có bình luận nào.