您的位置:首页 > 其它

HDU 1421 搬寝室 简单DP

2017-07-18 11:46 253 查看
搬寝室是很累的,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


看到这种题就应该想到动态规划 。。。。

我还是太菜了

一共最多有2000个物品,所以最高复杂度应该为O(n^2).

那就首先不用考虑区间DP,首先看能不能将其转化为背包问题。

设DP[i][j] 为前i件物品凑j对后所需要的最小疲劳

显然这么定义是可行的。

那么DP[i][j]=min(dp[i][j],dp[i-2][j-1]+pow(差值最小对,2));

这个差值最小对怎么弄呢。

这里就需要贪心弄一下,我们将物品sort排下序,那么总是取相邻的物品凑成一对肯定是最优的。

#include <iostream>
#include<queue>
#include<algorithm>
#include<cstdio>
#include<string>
#include<cmath>
#include<string.h>
using namespace std;
long long dp[2005][1005];
long long cas[2005];
#define INIF 2147483646
long long mypow(long long x)
{
return x*x;
}
int  main()
{
int n,k;
while(~scanf("%d %d",&n,&k))
{
for(int i=1;i<=n;i++)
{
scanf("%lld",&cas[i]);
}
sort(cas+1,cas+n+1);
for(int i=0;i<=n;i++)
for(int j=1;j<=k;j++)
dp[i][j]=INIF;
dp[0][0]=0;
for(int len=2;len<=n;len++)
{
for(int ks=1;ks<=len/2;ks++)
{
dp[len][ks]=min(dp[len-2][ks-1]+mypow(cas[len-1]-cas[len]),dp[len-1][ks]);
}
}
cout<<dp
[k]<<endl;
}
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: