搬寝室 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; }