农 场 主 农场主 农场主
题目
小张是一个养马农场的农场主,他要把N只马分配到K个马房里,放置的规则是:第1 到 第Pi只马放入第一个马房,第Pi+1 到第Pk只放入第二个马房,…以此类推。此外对于每一个马房都有一个叫做“不高兴系数”,即白色马的数量*黑色马的数量。你的任务是合理地分配这N只马,使得它所有马房的“不高兴系数”和最小。
输入
从文件 f a r m e r . i n farmer.in farmer.in中读入数据,文件中第一行有 2 2 2 个整数: $N ( 1 <= N <= 500 ) $和 K ( 1 < = K < = N ) K ( 1 <= K <= N) K(1<=K<=N)
接下来的N行有N个数。第 I 行为第 I 只马的颜色: 1 是黑色, 0 是白色。
输出
将结果输出到文件 f a r m e r . o u t farmer.out