题目链接: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
Input
Output
Sample Input
2 1 1 3
Sample Output
4
Author
Source
Recommend
思路:首先想一下每一对该如何取之。假设有四件物品,重量分别为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;
}