COGS 384. 筷子

Posted 2020-09-16 Alex丶Baker

★☆   输入文件：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

 

思路：一个筷子有3种状态，就是什么都不选，选前面的，选后面的

于是  选前面的筷子可以看做前面的筷子选后面的筷子

于是就剩下2中状态了

f[i][j]表示第i个筷子，前面已经选了j对，的最小的差的平方和

 

#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#define maxn 1000010
#define maxm 1010
using namespace std;
int n,k,l[maxn],minn,f[maxm][maxm],jud;
bool vis[maxn];
int main()
{
    //freopen("chop.in","r",stdin);
    //freopen("chop.out","w",stdout);
    memset(f,0x7f,sizeof(f));
    f[0][0]=0;
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++) scanf("%d",&l[i]),jud++;
    if(jud<2*k+6)
    {
        cout<<"-1"<<endl;
        return 0;
    }
    else {
        sort(l+1,l+n+1);
        for(int i=2;i<=n;i++)
            for(int j=1;j<=k+3;j++)
                f[i][j]=min(f[i-1][j],f[i-2][j-1]+(int)pow((l[i]-l[i-1]),2));
        printf("%d",f[n][k+3]);
    }
    return 0;
}