Despotcovit

Time limit: 1s Memory limit: 32MB Input: despotcovit.in Output: despotcovit.out

Ion Piețaru’ are o pasiune imensă pentru cai. În fiecare zi, iese cu ei la plimbare și seara îi aduce înapoi la grajduri. Problema lui cea mai mare însă, este cum să facă acest lucru.

Ion obișnuiește să îi poziționeze mereu într-o linie dreaptă pe toți, iar pentru că își iubește caii și nu vrea să îi obosească, vine cu o modalitate de a îi așeza astfel încât să nu implice prea multă mutare din partea lor. Astfel, Ion pune primii k1k_1 cai în grajdul 11, următorii k2k_2 cai în grajdul 22 și așa mai departe. Pe langă asta, el își dorește ca niciun grajd să nu ramână liber din toate cele GG și niciun cal să nu doarmă afară.

Caii lui Ion Piețaru’ sunt de 22 culori, albi si albaștri. Caii de culori diferite nu se înteleg deloc, așa că un grajd cu ii cai albaștri și jj cai albi va duce la o tristețe de i×ji \times j. Tristețea totală a unei așezări în grajduri va fi suma tristeților pentru toate cele GG grajduri. Ion se bazează pe tine să îl ajuți cu așezarea cailor.

Cerință

Cunoscând NN si GG, tu trebuie să găsești modalitatea optimă de a pune cei NN cai în cele GG grajduri astfel încât tristețea totală să fie minimă.

Date de intrare

Pe prima linie a fișierului de intrare despotcovit.in se găsesc două numere întregi, NN și GG.
Pe următoarele NN linii, se găseste câte un număr 00 sau 11 care reprezintă culoarea calului (11 pentru albastru, 00 pentru alb).

Date de ieșire

Pe prima linie a fișierului de ieșire despotcovit.out se va găsi un singur număr întreg, reprezentând valoarea minimă posibilă a tristeții totale.

Restricții și precizări

  • 1N5001 \leq N \leq 500;
  • 1GN1 \leq G \leq N;
  • Pentru teste în valoare de 2020 de puncte, N50N \leq 50.

Exemplu

despotcovit.in

6 3
1
1
0
1
0
1

despotcovit.out

2

Explicație

Plasăm primii 22 cai in grajdul 11 (tristețe 00), următorii 33 in grajdul 22 (tristețe 22), iar ultimul cal în grajdul 33 (tristețe 00).

Log in or sign up to be able to send submissions!