先排序。因为相邻的两个数平方肯定是最小的。
然后一个一个地增长DP。判断是否要保留。
DP要注意枚举状态,把全部状态做出来,我们分析的时候容易从一方面切入,但是我们要后面的实现时要全面的枚举。
这题还可以优化一下空间,要滚动数组。
#include <iostream>
#include <string.h>
#include <algorithm>
using namespace std;
int dp[2001][1002];
int a[2002];
int main()
{
int n,k,i,j;
while(cin>>n>>k)
{
for(i = 1;i <= n;i ++)
cin>>a[i];
a[0] = 0;
sort(a,a+n+1);
memset(dp,0,sizeof(dp));
for(i = 1;i <= n;i ++)
{
for(j = 1;j<=k&&j*2 <= i;j ++)
{
if(j*2 == i)
dp[i][j] = dp[i-2][j-1] + (a[i] -a[i-1])*(a[i]-a[i-1]);
else
dp[i][j] = min(dp[i-2][j-1] + (a[i] -a[i-1])*(a[i]-a[i-1]),dp[i-1][j]);
}
}
cout<<dp[n][k]<<endl;
}