COGS 384.筷子

384. 筷子

★☆   输入文件：chop.in   输出文件：chop.out   简单对比
时间限制：1 s   内存限制：128 MB

A先生有很多双筷子。确切的说应该是很多根，因为筷子的长度不一，很难判断出哪两根是一双的。这天，A先生家里来了K个客人，A先生留下他们吃晚饭。加上A先生，A夫人和他们的孩子小A，共K+3个人。每人需要用一双筷子。A先生只好清理了一下筷子，共N根，长度为T1,T2,T3,……,TN.现在他想用这些筷子组合成K+3双，使每双的筷子长度差的平方和最小。(怎么不是和最小？？这要去问A先生了，呵呵)

输入

输入文件共有两行，第一行为两个用空格隔开的整数，表示 N,K(1≤N≤100, 0<K<50)，第二行共有N个用空格隔开的整数，为Ti.每个整数为1～50之间的数。

输出

输出文件仅一行。如果凑不齐 K+3双，输出-1，否则输出长度差平方和的最小值。

样例

chop.in

10 1

1 1 2 3 3 3 4 6 10 20

 

chop.out

5

说明

第一双 1 1

第二双 2 3

第三双 3 3

第四双 4 6

(1-1)^2+(2-3)^2+(3-3)^2+(4-6)^2=5

题目大意：有m+3个人，n双筷子。如果n双筷子不够m+3个人用输出-1，否则

求一个搭配，使得成双的筷子的高度差的平方的和最小。

题解：dp

f[i][j]=min(f[i-1][j],f[i-2][j-1]+(a[i]-a[j])^2)。当前筷子选还是不选。

代码

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m,a[120],f[120][120];
long long ans;
int main(){
    freopen("chop.in","r",stdin);
    freopen("chop.out","w",stdout);
    scanf("%d%d",&n,&m);m+=3;
    if(m*2>n){
        printf("-1");return 0;
    }
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    sort(a+1,a+n+1);
    memset(f,0x3f3f,sizeof(f));
    f[0][0]=0;
    for(int i=2;i<=n;i++)
     for(int j=1;j<=m;j++)
       f[i][j]=min(f[i-1][j],f[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
    printf("%d\n",f[n][m]);
    fclose(stdin);
    fclose(stdout);
    return 0;
}