首先可以看出排序之后,最优解肯定是每一对都相邻才是最优的
那么我们就要找构成最优解的相邻组
设f[i][j]是前i个字符,k对的最小值
如果当前这个筷子不取的话,f[i][j] = f[i-1][j]
如果取的话 f[i][j] = f[i-2][j-1] + (a[i]-a[i-1])*(a[i]-a[i-1])
取最小值就好了。
#include
#include
#include
#include
#define REP(i, a, b) for(int i = (a); i < (b); i++)
using namespace std;
const int MAXN = 112;
int f[MAXN][MAXN], a[MAXN], n, k;
int main()
{
scanf("%d%d", &n, &k);
k += 3;
if(k * 2 > n) { puts("-1"); return 0; }
memset(f, 63, sizeof(f));
REP(i, 1, n + 1) scanf("%d", &a[i]), f[i][0] = 0;
REP(i, 0, n + 1) f[i][0] = 0;
sort(a + 1, a + n + 1);
REP(i, 2, n + 1)
REP(j, 1, k + 1)
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][k]);
return 0;
}