HDU 1421 搬寝室

题目链接：http://acm.split.hdu.edu.cn/showproblem.php?pid=1421

搬寝室

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)

Total Submission(s): 25720    Accepted Submission(s): 8843

Problem Description

搬寝室是很累的,xhd深有体会.时间追述2006年7月9号,那天xhd迫于无奈要从27号楼搬到3号楼,因为10号要封楼了.看着寝室里的n件物品,xhd开始发呆,因为n是一个小于2000的整数,实在是太多了,于是xhd决定随便搬2*k件过去就行了.但还是会很累,因为2*k也不小是一个不大于n的整数.幸运的是xhd根据多年的搬东西的经验发现每搬一次的疲劳度是和左右手的物品的重量差的平方成正比(这里补充一句,xhd每次搬两件东西,左手一件右手一件).例如xhd左手拿重量为3的物品,右手拿重量为6的物品,则他搬完这次的疲劳度为(6-3)^2 = 9.现在可怜的xhd希望知道搬完这2*k件物品后的最佳状态是怎样的(也就是最低的疲劳度),请告诉他吧.

 

Input

每组输入数据有两行,第一行有两个数n,k(2<=2*k<=n<2000).第二行有n个整数分别表示n件物品的重量(重量是一个小于2^15的正整数).

 

Output

对应每组输入数据,输出数据只有一个表示他的最少的疲劳度,每个一行.

 

Sample Input

  
  

   
   2 1
1 3
  
  

 

Sample Output

  
  

   
   4
  
  

 

Author

xhd

 

Source

ACM暑期集训队练习赛（二）

 

Recommend

lcy

思路：首先想一下每一对该如何取之。假设有四件物品，重量分别为a，b，c，d，不妨设四个重量就是按照从小到大顺序排列的，则我们取一对有三种情况：ab和cd，ac和bd，ad和bc。这样各自的疲劳度分别只与ab+cd，ac+bd，ad+bc有关，且要使得疲劳度最小，显然相关量要最大，这样就能够知道第一种最大，即取的时候一定是取两个重量相邻的物件。其次我们再来考虑状态转移方程，设dp[i][j]表示前i件物品里面取j对得到的最小疲劳度，则状态转移方程为：

dp[i][j] = min(dp[i-1][j], dp[i-2][j-1]+squ(num[i-1]-num[i-2]))

，其中squ是求一个数的平方，num[i]表示第i+1件物品的重量。最后考虑一下边界情况，dp[2][1] = squ(num[1]-num[0])，dp[i][i>>1] = dp[i-2][(i>>1)-1]+squ(num[i-1]-num[i-2])(i为偶数)，为什么呢？因为偶数时，没法选择，必须全选。详见代码。

附上AC代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 2005;
int num[maxn];
int dp[maxn][maxn>>1];
int n, k;

int squ(int x){
	return x*x;
}

int main(){
	while (~scanf("%d%d", &n, &k)){
		for (int i=0; i<n; ++i)
			scanf("%d", num+i);
		sort(num, num+n);
		dp[2][1] = squ(num[1]-num[0]);
		for (int i=3; i<=n; ++i){
			for (int j=1; j<=(i>>1); ++j)
				dp[i][j] = min(dp[i-1][j], dp[i-2][j-1]+squ(num[i-1]-num[i-2]));
			if ((~i) & 1)
				dp[i][i>>1] = dp[i-2][(i>>1)-1]+squ(num[i-1]-num[i-2]);
		}
		printf("%d\n", dp[n][k]);
	}
	return 0;
}