Description
搬寝室是很累的,xhd深有体会.时间追述2006年7月9号,那天xhd迫于无奈要从27号楼搬到3号楼,因为10号要封楼了.看着寝室里的n件物品,xhd开始发呆,因为n是一个小于2000的整数,实在是太多了,于是xhd决定随便搬2k件过去就行了.但还是会很累,因为2k也不小是一个不大于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,刷了很多题之后看到这个仍旧没有思路,看了题解之后有种原来还可以这样的感觉,记录下来多思考思考。
刚开始我毫无头绪在于:
- 初始状态可以设置为只有2个东西且选2个。但其实DP的初始状态完全可以设置为不会发生的情况 – dp[0][j],j个中选0个,那结果肯定是0
- 从2个中选,到从3个中选不会推。但是DP的状态推导一定要根据dp[][]后的所有参数来的,像这种就是dp[i][j],要通过一对i,j推,两个i之间肯定是转化不了的。所以这个问题应该理解成,怎么从2个中选2个,转移到3个中选2个。这样问题就明朗了些(归根结底可能就是状态设定经验不够)
思路
这类DP的特点是,你要先得出某个结论之后,才能设计出合理的状态
- 首先肯定是得把给的物品按质量排好序
- 选取哪些个物品
- 在所选取的物品中进行两两配对
第1步排好序之后,就是怎么选择的问题了,关于这个怎么选择,首先要明确一个原则:
- 假如有a>b>c>d,那么配对方式一定是ab,cd,证明:
这种配对方法的疲劳度是(a-b)2+(c-d)2,还有一种是(a-c)2+(b-d)2,将二者相减结果为2(a-d)(c-b)<=0,就可以推出前者<=后者,推导就是化简即可,在此不赘述。
这说明若是选出来abcdef…配对方法一定是ab,cd,ef… - 了解了配对方法之后我们再看选择的问题:由于要满足疲劳度最小,在选出来的值中配对要找相邻的,同时,在选择的时候也得找相邻的。
例如按大小x>y>z(假设xyz是前面选完剩下的),那肯定是得选xy或者yz呗
说这么多就是为了证明一点,肯定一选就是一对邻居
这样状态的转移你可能就有些思路了:
dp[i][j]表示:在前j个物品中选择i对
则这种情况就可以分为:选的第i对,是不是j-1,j。
- 若是则dp[i][j] = dp[i-1][j-2]+ j-1,j的疲劳度;
- 若不是,则j物品一定没有被选择,(因为在这种情况下,j只有j-1一个邻居)
dp[i][j] = dp[i][j-1];
本题的状态初始化也要注意,有两种情况的初始化
- 对于在任意j中取i = 0,dp[0][j] = 0;
- 对于2*i >= j的情况,dp[i][j] = inf;
初始化值不同的原因是转移方程是取min,这就像一颗解答树,dp[0][j]虽然不符合题意,但是其上可以长出其它解(后面状态可以由它+一个平方得到)。而第二种情况,是根本不可能存在,所以不能给它被后面利用的机会。
代码
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
long long a[2010];
long long dp[1010][2010];
int main(void)
{
//freopen("shuru.txt","r",stdin);
int n,k;
while(scanf("%d%d",&n,&k)!=EOF)
{
a[0] = 0;
for(int i = 1;i<=n;i++) scanf("%lld",&a[i]);
sort(a,a+n+1);//竟然折在sort上了,sort(a,a+n)少排序了一个!!!
memset(dp,0x3f,sizeof(dp));//dp初始化,因为2*i>=j情况比较多,所以先初始化这种情况的
for(int j = 0;j<=n;j++) dp[0][j] = 0;
for(int i = 1;i<=k;i++)
{
for(int j = 2*i;j<=n;j++)
{
dp[i][j] = min(dp[i-1][j-2]+(a[j]-a[j-1])*(a[j]-a[j-1]),dp[i][j-1]);
}
}
printf("%lld\n",dp[k][n]);
}
return 0;
}