码迷,mamicode.com
首页 > 其他好文 > 详细

!HDU 1500 Chopsticks-dp-(分组问题)

时间:2015-07-13 20:41:11      阅读:133      评论:0      收藏:0      [点我收藏+]

标签:hdu   dp   

题意:一种新式筷子:每副有三支,两支短的,长度差要尽可能小;一支长的,只要满足是一副里最长的就行。在n支筷子里选k副,要求每副里的短筷子的长度差总和最小

分析:

分组dp模型。这题跟搬寝室那题差不多,但这题的每组里多了一个长筷子,所以排序的时候从大到小,这样才能在一遍dp的时候一遍把长的选进每一组里,具体怎么实现的不要过于钻牛角尖。

正如搬寝室里总结的分组模型的实现方式有两种,这里也给出两个代码。注意具体实现上的一些细节差异,还有初始化。

把两道题结合起来看加深理解。

代码1:

#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#define min(a,b) a<b?a:b;
#define INF 1000000007
using namespace std;
int t,k,n;
int dp[5010][1100],a[5100];
bool cmp(int i,int j)
{
	return i>j;
}
int main()
{
	cin>>t;
	while(t--){
		cin>>k>>n;
		for(int i=1;i<=n;i++) cin>>a[i];
		sort(a+1,a+n+1,cmp);
		k+=8;
		for(int i=0;i<=n;i++)
		  for(int j=1;j<=k;j++)
		     dp[i][j]=INF;
		for(int i=0;i<=n;i++) dp[i][0]=0;		
		for(int i=3;i<=n;i++){
			for(int j=1;j*3<=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]));
			}
		}
		cout<<dp[n][k]<<endl;
	}
}
代码2:

#include<iostream>
#include<algorithm>
#include<cstring>
#define min(a,b) a<b?a:b;
using namespace std;
int t,k,n;
int dp[1010][5100],a[5100];
bool cmp(int i,int j)
{
	return i>j;
}
int main()
{
	cin>>t;
	while(t--){
		cin>>k>>n;
		for(int i=1;i<=n;i++) cin>>a[i];
		sort(a+1,a+n+1,cmp);
		memset(dp,0,sizeof(dp));
		k+=8;
		for(int i=1;i<=k;i++){
			dp[i][3*i]=dp[i-1][3*i-2]+(a[3*i]-a[3*i-1])*(a[3*i]-a[3*i-1]);
			for(int j=3*i+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]));
		}
		cout<<dp[k][n]<<endl;
	}
}



版权声明:本文为博主原创文章,未经博主允许不得转载。

!HDU 1500 Chopsticks-dp-(分组问题)

标签:hdu   dp   

原文地址:http://blog.csdn.net/ac_0_summer/article/details/46866677

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!