题目大意
现有 n n n 根筷子,第 I I I 根筷子长度为 a i a_i ai,现要组成 k + 3 k+3 k+3 双筷子,先将 x x x 定义为组成的 k + 3 k+3 k+3 双中每一双中的两根筷子的长度差的平方的和,试求出 x x x 的最小值。
对于 100 % 100\% 100% 的数据,满足 1 ≤ n ≤ 100 , 0 < k < 50 , 1 ≤ a i ≤ 50 1 \leq n \leq 100,0 < k < 50, 1 \leq a_i \leq 50 1≤n≤100,0<k<50,1≤ai≤50。
解题思路
显然的 DP
。
先将 a a a 从小到大排序。
以得,选相邻的筷子是最优的。
记 f i , j f_{i,j} fi,j 为前 i i i 根筷子中选 j j j 双筷子的最小 x x x 值。
AC CODE
#include <bits/stdc++.h>
#define INF 0x3f3f3f3f
using namespace std;
int n, k;
int a[101], f[101][101];
signed main()
{
scanf("%d%d", &n, &k);
k += 3;
for(int i = 1; i <= n; ++i) scanf("%d", &a[i]);
sort(a + 1, a + n + 1);
if(k * 2 > n)
{
printf("-1\n");
return 0;
}
for(int i = 0; i <= n; ++i)
{
for(int j = 1; j <= k; ++j)
f[i][j] = INF;
}
for(int i = 2; i <= n; ++i)
for(int j = 1; j <= k; ++j)
f[i][j] = min(f[i - 2][j - 1] + (a[i] - a[i - 1]) * (a[i] - a[i - 1]), f[i - 1][j]);
printf("%d", f[n][k]);
return 0;
}