CSES - Koulut
  • Time limit: 2.00 s
  • Memory limit: 512 MB

Tien varrella on nn taloa, ja tiedät jokaisesta talosta, montako lasta siinä asuu. Tehtäväsi on perustaa kk koulua niin, että jokainen koulu on jossakin talossa ja lasten yhteismatka kouluun on mahdollisimman lyhyt.

Syöte

Syötteen ensimmäisellä rivillä on kaksi kokonaislukua nn ja kk: talojen määrä ja koulujen määrä. Talot on numeroitu 1,2,,n1,2,\ldots,n.

Sitten syötteessä on nn kokonaislukua t1,t2,,tnt_1,t_2,\ldots,t_n: montako lasta kussakin talossa asuu.

Koulujen perustamisen jälkeen kaikki tietyssä talossa asuvat lapset menevät lähimpään kouluun. Matkan kustannus on lasta kohden ab|a-b|, missä aa on talon numero ja bb on koulun numero.

Tuloste

Tulosta yksi kokonaisluku: pienin mahdollinen kaikkien lasten yhteiskoulumatka.

Rajat

  • 1n,k50001 \le n,k \le 5000
  • 1ti1061 \le t_i \le 10^6

Esimerkki

Syöte:

6 2
2 7 1 4 6 4

Tuloste:

11

Selitys: koulut perustetaan taloihin 2 ja 5.