搬寝室
时间限制: 1 Sec 内存限制: 128 MB题目描述
搬寝室是很累的,小明深有体会.那天小明迫于无奈要从4号楼搬到5号楼,因为4号要封楼了.看着寝室里的n件物品,小明开始发呆,因为n是一个小于2000的整数,实在是太多了,于是小明决定随便搬2*k件过去就行了.但还是会很累,因为2*k也不小是一个不大于n的整数.幸运的是小明根据多年的搬东西的经验发现每搬一次的疲劳度是和左右手的物品的重量差的平方成正比(这里补充一句,xhd每次搬两件东西,左手一件右手一件).例如小明左手拿重量为3的物品,右手拿重量为6的物品,则他搬完这次的疲劳度为(6-3)^2 = 9.现在可怜的小明希望知道搬完这2*k件物品后的最佳状态是怎样的(也就是最低的疲劳度),请告诉他吧.
输入
每组输入数据有两行,第一行有两个数n,k(2<=2*k<=n<2000).第二行有n个整数分别表示n件物品的重量(重量是一个小于2^15的正整数).
输出
对应每组输入数据,输出数据只有一个表示他的最少的疲劳度,每个一行.
样例输入
2 1
1 3
样例输出
4
解题思路:一道DP题
给定n个物品,每个物品有重量,
从中选出m对,使得这m对物品重量差的平方和最小。
疲劳度:m对物品重量差的平方和
分析与解题思路
先对n中物品的重量排序
令dp[i][j]表示前i个物品中选j对的最小疲劳度。
则dp[i][j]可能含有第i个物品(这种情况下,第i种物品一定是和第i-1个物品配对),
则dp[i][j]=dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1])
dp[i][j]的j对也可能不含有第i个物品,此时有
dp[i][j]=dp[i-1][j]
状态转移方程
dp[i][j]=min{dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1]),dp[i-1][j]
#include <stdio.h>
#include <stdlib.h>
#define size 2005
#define INIT 2147483646
int cmp(const void *a,const void *b)
{
return *(int *)a-*(int *)b;
}
int Min(int a,int b)
{
return a<b?a:b;
}
int dp[size][1005];
int main()
{
int n,k,i,j;
int val[size] = {0};
dp[0][0] = 0;
while (scanf("%d%d",&n,&k)!=EOF)
{
val[0] = 0;
for(i=1;i<=n;i++)
scanf("%d",&val[i]);
qsort(val+1,n,sizeof(val[0]),cmp);
for (i=0;i<=n;i++)
{
for(j=1;j<=k;j++)
dp[i][j] = INIT;
}
for (i=2;i<=n;i++)
{
for (j=1;j*2<=i;j++)
dp[i][j] = Min(dp[i-2][j-1]+(val[i]-val[i-1])*(val[i]-val[i-1]),dp[i-1][j]);
}
printf("%d\n",dp[n][k]);
}
return 0;
}