题目链接:
UVALive 6434 Number Assignment
题意:
将n个数分成m堆使得每堆的最大值和最小值之差的和最小,输出最小差和。n,m<=100.
分析:
先将整个数组排序。dp[i][j]表示将前i个数分成j堆的最小差和。
初始化:dp为INF,但是dp[0][j]为0。
状态转移方程:
考虑第i个数所在堆的最小值。那么剩下的数分成j-1堆,又因为数据是从小到大排序的,所以dp[i][j]=min(dp[k-1][j-1]+data[i]-data[k]);其中1<=k<=i。
状态转移方程中也包括了第i个数自成一堆的情况(k=i)。
//0K 26MS
#include <cstdio>
#include <cstring>
#include <cmath>
#include <algorithm>
#include <climits>
using namespace std;
const int MAX_N=110;
const long long INF=LONG_LONG_MAX/2;
int T,n,m,cases=0;
long long data[MAX_N],dp[MAX_N][MAX_N];
void solve()
{
for(int i=1;i<=n;i++) {
for(int j=0;j<=m;j++) {
dp[i][j]=INF;
}
}
memset(dp[0],0,sizeof(dp[0]));
for(int i=1;i<=n;i++){
for(int j=1;j<=m;j++){
for(int k=1;k<=i;k++){
dp[i][j]=min(dp[k-1][j-1]+data[i]-data[k],dp[i][j]);
}
}
}
printf("Case #%d: %lld\n",++cases,dp[n][m]);
}
int main()
{
//freopen("Ain.txt","r",stdin);
scanf("%d",&T);
data[0]=0;
while(T--){
scanf("%d%d",&n,&m);
for(int i=1;i<=n;i++) {
scanf("%lld",&data[i]);
}
sort(data+1,data+n+1);
solve();
}
return 0;
}