每天晚上,牧马人把所有马排在一条直线上带回马厩。由于马儿们很累,牧马人尽量不让它们移动。所以他发明了一个算法:他把前P1匹马放在第一个马厩,后P2匹马放在第二个马厩……而且,他不想K个马厩任何一个是空的,并且没有马被留在外边。牧马人有黑白两种颜色的马,两种颜色的马相处不好。如果有i匹黑马和j匹白马同在一个马厩中,那么这个马厩的忧愁系数为i×j,总忧愁系数是所有马厩的马的忧愁系数的和。忧愁系数过大会表现在马儿互相打架或者马儿彻夜长嘶,请想方法把N匹马放进K个马厩使得总忧愁系数最小。
第一行是N(1\le N\le500)和K(1\le K\le N),下面N行描述每匹马的颜色,1代表黑色,0代表白色。
最小的总忧愁系数。
6 3 1 1 0 1 0 1
2
将前2匹马放在第一个马厩,随后3匹马放在第二个马厩,最后一匹马放在第三个马厩。