搬寝室(动态规划)
搬寝室
Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 20507 Accepted Submission(s):
6973
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
解析:
很明显该题中要取得两件物品和是相邻的,所以要先对物品重量进行排序,然后取差的平方。
a[i]表示搬第i对物品时产生的疲劳度,而f[i][j]表示在i件物品中去j对产生的最小疲劳度,即最佳状态。
- 第n件物品不搬,即在前n - 1件物品中搬k对,那么疲劳值仍为f[n - 1][k];
- 第n件物品要搬,那么根据上面所证,第n - 1件物品也要同时搬,即在前n - 2件物品中搬k - 1对物品,再搬最后一对物品,那么疲劳值为f[n - 2][k - 1] + a[n - 1],n - 1是因为对数必然比总物品数少1。
1. f[n][k] = min(f[n - 1][k], f[n - 2][k - 1] + a[n - 1])
应该注意的是要考虑边界问题,f[i][j]中:
- 当2 * j > i时,即要搬的数量超过了物品总量,这是不可能发生的,因此此时令f[i][j]为无穷大;
- 当j == 0时,即在一对物品都没搬时,所需疲劳值应该是0,此时令f[i][j] = 0。
1 #include <iostream> 2 #include <cstdio> 3 #include <algorithm> 4 using namespace std; 5 int f[2010][2010]; 6 int getf(int i,int j) 7 { 8 if(j*2>i)//要搬的数目大于物品总数 9 return 1000000000;//返回一个很大的数,而min函数绝对不可能取到 10 if(j==0)//要搬的数为0,那么疲劳度为0 11 return 0; 12 return f[i][j];//正常情况 13 } 14 int main() 15 { 16 int n,k,i,j; 17 while(cin>>n>>k) 18 { 19 f[2010][2010]=0; 20 int a[2010],sum=0; 21 for(i=1;i<=n;i++) 22 cin>>a[i]; 23 sort(a+1,a+n+1); 24 for(i=1;i<n;i++) 25 { 26 a[i]=a[i+1]-a[i]; 27 a[i]*=a[i]; 28 } 29 for(j=1;j<=k;j++) 30 for(i=2*j;i<=n;i++) 31 f[i][j]=min(getf(i-1,j),getf(i-2,j-1)+a[i-1]);//状态转移方程 32 cout<<f[n][k]<<endl; 33 } 34 }