题意:
小Q同学为了准备今年的ICPC Regional,计划在m天之内刷掉n道题,每道题有一个难度值,其中第i道题的难度值为a[i]。
然而处于半颓废状态中的小Q同学不希望在同一天中做难度差距悬殊的题目,定义第i天中刷的题的难度的最大值减最小值为d[i](如果第i天没有刷题,则d[i]=0),那么整个计划的难度为 。
然而处于半颓废状态中的小Q同学不希望在同一天中做难度差距悬殊的题目,定义第i天中刷的题的难度的最大值减最小值为d[i](如果第i天没有刷题,则d[i]=0),那么整个计划的难度为 。
小Q同学可以按照任意的顺序刷题,并且一天中可以刷任意多道题,但是每道题只需要做一次,现在小Q同学想知道完成这个计划的总难度的最小值是多少。
思路:暴力三重for循环dp所有可能结果。
#include <stdio.h>
#include <iostream>
#include <string.h>
#include <algorithm>
#define siz 505
#define LL long long
using namespace std;
int n,m;
int gp[siz];
LL dp[siz][siz];
void solve()
{
LL x;
for(int i=1; i<=n; i++)
{
x = gp[i] - gp[1];
dp[1][i] = x * x;
}
for(int i=2; i<=m; i++)
{
for(int j=1; j<=n; j++)
{
dp[i][j] = 1ll*(gp[j] - gp[1]) * (gp[j] - gp[1]);
for(int k=2; k<=j; k++)
{
x = gp[j] - gp[k];
dp[i][j] = min(dp[i-1][k-1] + x * x,dp[i][j]);
}
}
}
printf("%lld\n",dp[m][n]);
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d%d",&n,&m);
for(int i=1; i<=n; i++)
{
scanf("%d",&gp[i]);
}
if(n<=m)
{
printf("%d\n",0);
continue;
}
sort(gp+1,gp+n+1);
solve();
}
}