#include "iostream"
#include "algorithm"
#include "cstring"
#define INF 99999999
using namespace std;
int comp(const void *a,const void *b);
int dp[2005][1005] ;
int main(){
int n, k;
int w[2005];
memset(w,0,sizeof(w));
while(scanf("%d%d",&n,&k) != EOF){
for( int i = 1; i <= n; i++){
scanf("%d",&w[i]);
}
qsort(&w[1] ,n,sizeof(w[0]),comp);
for(int i=1;i<=n;i++)
for(int j=1;j<=k;j++)
dp[i][j]=0xfffffff;
for(int i = 0; i <= n; i++)
dp[i][0] = 0;
for ( int i = 2; i<= n ; i++)
for ( int j = 1; j*2 <= i; j++){
dp[i][j] = min(dp[i-1][j],dp[i-2][j-1] + (w[i]-w[i-1])*(w[i]-w[i-1]));
}
printf("%d\n",dp[n][k]);
}
}
int comp(const void *a,const void *b){
return *(int*)a - *(int *)b;
}
在本题的学习中,思路是很清晰的,但是过程中语法点也确实让我发现之前的疏漏。
1. memset是按字节填充,memset(dp,99999999,sizeof(dp))是绝对不行的,赋值成了-1,可以使用memset(dp,0x3f,sizeof(dp))进行赋最大值,或者直接两个for循环得了。
2. qsort的函数配置,因为sort容易算错开始和末尾,不如使用qsort快捷方便,同时,comp函数记得写。