首页 > 代码库 > [HDU 1421]搬寝室(富有新意的DP)
[HDU 1421]搬寝室(富有新意的DP)
题目地址:http://acm.hdu.edu.cn/showproblem.php?pid=1421
题目大意:给出n个物品,要你选出k对物品,使得每对物品之差的平方之和最小。
思路;动态规划求解,首先将物品按重量升序排序,根据贪心思想,每对物品必然是相邻的两个物品,这样重量差的平方才能保证最小。然后DP,用f[i][j]表示前i个物品,选j对的最少代价,这里需要进行一个重要的分类讨论:
(1)如果j*2=i,即前i个物品全部用于配对了,很明显第j对物品是第i-1号物品和第i号物品,f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])^2
(2)其他情况。前i个物品并没有完全用于配对,可以选择第i个物品不配对,这时f[i][j]=f[i-1][j],或者第i个物品参与配对,同(1),第i个物品和第i-1个物品配对,f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])^2
最终的答案就是f[n][k]
#include <iostream> #include <stdio.h> #include <stdlib.h> #include <string.h> #include <algorithm> #define MAXN 2010 using namespace std; int f[MAXN][MAXN]; //f[i][j]=前i个物品,选j对的最小花费 int w[MAXN]; int min(int a,int b) { if(a<b) return a; return b; } int main() { int n,k; while(scanf("%d%d",&n,&k)!=EOF) { memset(f,0,sizeof(f)); w[0]=0; for(int i=1;i<=n;i++) scanf("%d",&w[i]); sort(w+1,w+n+1); for(int i=2;i<=n;i++) //前i个物品 for(int j=1;j*2<=i;j++) //选j对 { if(i!=2*j) f[i][j]=min(f[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]),f[i-1][j]); //要么第i个物品和第i-1个物品配对,要么留着第i个物品不配对 else f[i][j]=f[i-2][j-1]+(w[i]-w[i-1])*(w[i]-w[i-1]); } printf("%d\n",f[n][k]); } return 0; }
[HDU 1421]搬寝室(富有新意的DP)
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。