！HDU 4121 搬寝室-dp-（分组问题）

题意：共有n件物品，需要搬走2*k件，每次左右手各搬一件，疲劳度是（a-b）^2，求搬完k件最小的疲劳度和

分析：

也就是把n分成两两一组，共分k组。分组dp模型。

一般都是：dp[i][j]表示前i件分成j组的最优解；然后考虑第i件物品是否被选入：1).选：dp[i][j]=dp[i-2][j-1] ；2).不选：dp[i][j]=dp[i-1][j]。

综合：dp[i][j]=min(dp[i-2][j-1]+(a[i]-a[i-1])^2,dp[i-1][j])

有两种实现方法：

1.dp[i][j]表示前i件分j组：

memset(dp,INF,sizeof(dp);

for(int i=2;i<=n;i++)

    for(int j=1;j*2<=i;j++)

          dp[i][j]=min(dp[i-2][j-1]+(a[i]-a[i-1])^2,dp[i-1][j]);

2.dp[i][j]表示前j件分i组：

memset(dp,0,sizeof(dp));

for(int i=1;i<=k;i++){

    dp[i][i*2]=dp[i-1][i*2-2]+(a[i*2]-a[i*2-1])^2;

    for(int j=i*2+1;j<=n;j++)

    dp[i][j]=min(dp[i-1][j-2]+(a[j]-a[j-1])^2,dp[i][j-1]);

}

1的代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define INF 2147483646
using namespace std;
int n,k;
int dp[2010][1010];
int a[2010];
int min(int i,int j)
{
	return i<j?i:j;
}
int main()
{
	while(scanf("%d%d",&n,&k)!=EOF){
		for(int i=1;i<=n;i++) scanf("%d",&a[i]); 
		sort(a+1,a+n+1);
		for(int i=0;i<=n;i++) for(int j=1;j<=k;j++) dp[i][j]=INF;
		dp[0][0]=0;
		for(int i=2;i<=n;i++){
			for(int j=1;j*2<=i;j++){
				dp[i][j]=min(dp[i-1][j],dp[i-2][j-1]+(a[i]-a[i-1])*(a[i]-a[i-1]));
			}
		}
        printf("%d\n",dp[n][k]);
	}
}

2的代码：

#include<iostream>
#include<algorithm>
#include<cstring>
#include<cstdio>
#define INF 2147483646
using namespace std;
int n,k;
int dp[1010][2010];
int a[2010];
int min(int i,int j)
{
	return i<j?i:j;
}
int main()
{
	while(scanf("%d%d",&n,&k)!=EOF){
		for(int i=1;i<=n;i++) scanf("%d",&a[i]); 
		sort(a+1,a+n+1);
		memset(dp,0,sizeof(dp));
		for(int i=1;i<=k;i++){
			dp[i][i*2]=dp[i-1][2*i-2]+(a[2*i]-a[2*i-1])*(a[2*i]-a[2*i-1]);
			for(int j=i*2+1;j<=n;j++){
				dp[i][j]=min(dp[i][j-1],dp[i-1][j-2]+(a[j]-a[j-1])*(a[j]-a[j-1]));
			}
		}
        printf("%d\n",dp[k][n]);
	}
}