Description
小张是一个养马农场的农场主,他要把N只马分配到K个马房里,放置的规则是:第1 到 第Pi只马放入第一个马房,第Pi+1 到第Pk只放入第二个马房,……以此类推。此外对于每一个马房都有一个叫做“不高兴系数”,即白色马的数量*黑色马的数量。你的任务是合理地分配这N只马,使得它所有马房的“不高兴系数”和最小。
Input
从文件farmer.in中读入数据,文件中第一行有 2 个整数: N ( 1 <= N <= 500 ) 和 K ( 1 <= K <= N)。接下来的N行有N个数。第 I 行为第 I 只马的颜色: 1 是黑色, 0 是白色。
Output
将结果输出到文件farmer.out中,其结果为最小的“不高兴系数”的总和。
Sample Input
6 3
1
1
0
1
0
1
Sample Output
2
Source
elba
题解:
本题是动态规划(dp),用f[i,j]表示用i个马房装j匹马的最小不高兴系数
首先要预处理第i匹马时有多少只白马,多少只黑马,接着算出一个马房装1~n匹马的不高兴系数为多少。
如果马房=1,那么恭喜你