题意:
从n个数字中选取k对数,使得他们的差的平方和最小。
思路:
dp[i][j] 表示从前i个数字中选取j对数字的最优解。
1.选了第i个数字
dp[i][j] = dp[i-2][j-1] + (a[i-1] - a[i]) * (a[i-1] - a[i]);
2.没有选第i个数字
dp[i][j] = dp[i-1][j].
结果是2者选较小者。
代码:
#include <cstring>
#include <iostream>
#include <stdio.h>
#include <algorithm>
using namespace std;
int n,k;
const int maxn = 2010;
int a[maxn];
int dp[maxn][maxn];
int main() {
while(scanf("%d%d",&n,&k) != EOF) {
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
}
sort(a+1,a+1+n);
memset(dp,0x3f,sizeof(dp));
for(int i=0;i<=n;i++) {
dp[i][0] = 0;
}
for(int i=2;i<=n;i++) {
for(int j=1;j<=k;j++) {
dp[i][j] = min(dp[i-1][j],dp[i-2][j-1] + (a[i-1] - a[i]) * (a[i-1] - a[i]));
}
}
printf("%d\n",dp[n][k]);
}
}