搬寝室 1421

#include <iostream>
#include <algorithm>
using namespace std;
#define Min(a,b) (a < b ? a : b)
#define INF 2100000000
int dp[2001][1001];//第一个下标(2001)是指
int main() {
    int n;//物品的个数
	int k;//取多少对
	int a[2001];//数组a先用来各个物品的质量,再用来保存平方差。
	int i;//行
	int j;//列
	while(scanf("%d%d", &n, &k) != EOF){
		for(i = 1; i <= n; ++i){
			scanf("%d", &a[i]);
		}
		sort(a+1,a+1+n);//从小到大排序
		//算出相邻俩个数的平方差
		for(i = 1; i < n; ++i){
			//第一个平方差放在a[1],最后一个放在a[n-1];总共有n-1个平方差
			a[i] = (a[i+1] - a[i]) * (a[i+1] - a[i]);
		}

		/* dp数组 初始化 */
		for(i = 0; i <= n; i++)//0行到n行
            for(j = 1; j <= k; j++)//1列到k列
                dp[i][j] = INF;//初始化二维数组dp[i][j]
        for(i = 0; i <= n; i++) dp[i][0] = 0;//第一列初始化为0

		/* 公式:dp[i][j] = Min(dp[i-2][j-1]+a[i-1], dp[i-1][j]); */
		for(i = 2; i <= n; ++i){//注意这里i必须<=n
			//选取2对,3对,4对一直到j(2 * k <= n)对。
			for(j= 1; 2*j <= i; ++j){
				//dp[i][j]:前i个取j对
				//dp[i-2][j-1]+a[i-1]:前i-2个取j-1对,最后
				dp[i][j] = Min(dp[i-2][j-1]+a[i-1], dp[i-1][j]);//a[i-1]是最后一个平方差
				//dp[0][0]	1,1
			}
		}
		//总数n个,选取k对。
        printf("%d\n", dp[n][k]);
	}
    return 0;
}

猜你喜欢

转载自blog.csdn.net/Cannel_2020/article/details/7621913