题目大概意思是说给出k和n, 然后给出n根筷子从小到大给出, 现在要从这n根筷子中选出k + 8组筷子, 每组筷子包括三根, 现在要求所有k+8组每组筷子中最短两根筷子之差的平方的和最小,输出最小值。
DP
因为输入的筷子长度是有序的。所以一组筷子里面那两根小的一定是连续的两个。a[i]和a[i-1].
从大到小排序考虑。可以排除C的干扰。
#include <stdio.h>
#include <algorithm>
#define N 5005
using namespace std;
int k,n;
int a[N];
int dp[N][1010];
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&k,&n),k+=8;
for(int i=n;i>=1;i--)
scanf("%d",&a[i]),dp[0][i]=0;
for(int i=1;i<=n;i++)
for(int j=1;j<=k&&3*j<=i;j++)
{
dp[i][j] = dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]);
if(j*3!=i)
dp[i][j] = min(dp[i][j],dp[i-1][j]);
}
printf("%d\n",dp[n][k]);
}
return 0;
}