hdu 1421 搬寝室 题解（简单dp）

题目链接

题目大意

有n个行李，每个行李有一个重量。现在你要搬走2k个行李，你一共去k次。每次左手右手各拿一个行李，假设这两个行李的重量分别为x和y。那么这一次搬运产生的疲惫度是(x-y)*（x-y）现在你希望最小化疲惫度。2≤2k≤n<2000

题目思路

如果重量排序从小到大的四个物品a,b,c,d，选两对肯定是选（a,b）(c,d)
我本来以为是贪心，但是没有思路。其实确实不应该。
看到n=2e3，那么肯定就是o（n^2）的复杂度，很容易想到01背包的模型
设dp[i][j]表示前i个物品选j对，先sort一下，注意枚举以i=2开始，还有初始化要初始化所有dp[i][0]=0

代码

#include<cstdio>
#include<cstring> 
#include<algorithm> 
using namespace std;
typedef long long ll;
const int maxn=2e3+5;
int n,k,a[maxn],dp[maxn][maxn];
int main(){
	while(scanf("%d %d",&n,&k)!=-1){
		for(int i=1;i<=n;i++){
			scanf("%d",&a[i]);
		}
		sort(a+1,a+1+n);
		memset(dp,0x3f,sizeof(dp));//初始化最大值 
		for(int i=0;i<=n;i++){//初始化,注意是初始全部dp[i][0] 
			dp[i][0]=0;
		} 
		for(int i=2;i<=n;i++){//dp[i][j]表示前i个里面选j个 
			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]);	
	}
	return 0;
}