搬寢室
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
思路
dp[i][k]表示前i件物品中搬k對物品的最小值。不存在則爲無窮大。
狀態轉移方程爲:
dp[i][k]=min(dp[i-1][k],dp[i-2][k-2]+(a[i]-a[i-1])*(a[i]-a[i-1]));
錯誤代碼:
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=2005;
int a[N];
int dp[N][N];
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
memset(dp,0,sizeof(dp));//沒有正確初始化,不存在時要初始化爲inf
sort(a+1,a+n+1);
dp[2][1]=(a[2]-a[1])*(a[2]-a[1]);
for(int i=3;i<=n;i++)
{
for(int k=1;2*k<=i;k++)
{
dp[i][k]=min(dp[i-1][k],dp[i-2][k-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
}
}
printf("%d\n",dp[n][m]);
}
return 0;
}
修改版:
#include<cstdio>
#include<cstring>
#include<algorithm>
#define inf 1<<30
#define ll __int64
using namespace std;
const int N=2002;
int a[N];
int dp[N][N/2];
int main()
{
int n,m;
while(scanf("%d%d",&n,&m)!=EOF)
{
for(int i=1;i<=n;i++)
scanf("%d",&a[i]);
sort(a+1,a+n+1);
for(int i=1;i<=n;i++)
for(int k=1;k<=i;k++)
dp[i][k]=inf;
for(int i=0;i<=n;i++) dp[i][0]=0;
//初始化一定要正確
for(int i=2;i<=n;i++)
{
for(int k=1;2*k<=i;k++)
{
dp[i][k]=min(dp[i-1][k],dp[i-2][k-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
}
}
printf("%d\n",dp[n][m]);
}
return 0;
}