HDU 1421 搬寝室

/*
http://acm.hdu.edu.cn/showproblem.php?pid=1421

动态规划

无法用贪心！！！！

对于 这六个数 要搬3对 ：3 4 7 9 10 13： 如果用贪心，则是 3 4 然后 9 10 然后 7 13 总的疲劳值为 38
而 最小的 方案应该是 3 4 、 7 9 、 10 13 疲劳值为 14 

首先 对 物品的重量进行从小到大排序 

用 dp 二维数组 表示疲劳值 dp[i][j] 表示 对于前 i 个物品 搬运 j对时的最小疲劳值

对于 dp[i][j]  有两种情况： 1、搬运了 第i个物品 2、没有搬运 第 i 个物品

对于第一种情况，当搬运了 第 i 个物品则第 i-1 个物品也被一起搬运了，
（因为搬运相邻的物品的疲劳值最低，而 与 i 物品相邻的只要 i-1物品）

则：dp[i][j] = dp[i-2][j-1]+(thing[i]-thing[i-1])* (thing[i]-thing[i-1])

对于第二种情况， 没有搬运 第 i 个物品，则 dp[i][j] = dp[i-1][j]

所以 总的 dp[i][j] 的最小值是上面两种情况的较小值 



*/

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#define INIT 1000000000

int thing[2010];
int dp[2010][1005];

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 main()
{
	int n,k;
	while(scanf("%d %d",&n,&k)!=EOF)
	{
		memset(thing,0,sizeof(thing));
		memset(dp,0,sizeof(dp));
		int i;
		for(i=1; i<=n; i++)
			scanf("%d",&thing[i]);
		
		qsort(thing+1,n,sizeof(thing[0]),cmp);
		for(i=1; i<=n; i++)
		{
			int j;
			for(j=1; j<=k; j++)
				dp[i][j]=INIT;
		}
		for(i=1; i<=n; i++)
		{
			int j;
			for(j=1; j*2<=i; j++)
			{
				dp[i][j]=min(dp[i-2][j-1]+(thing[i]-thing[i-1])*(thing[i]-thing[i-1]),dp[i-1][j]);
			}
		}
		printf("%d\n",dp[n][k]);
	}
	return 0;
}