背包变式——能装的第K大值
hdu2639先吐槽坑点:n大于题目所讲100!!!
说下题意,给一个背包和一些物体的体积与价值求所能放的第K大价值。
很明显的是一个01背包变式,至于如何求一个第K大价值就是这一题的最大难点了。
首先讲一下第K大值的核心思想:第K大值必然出现在每一次的最大的前K个数字之间打一个比方也就是让你找全国第三大富翁怎么找?一个一个的?当然是把每一个省前三的富翁全部找出来比较一下就行了呀!!!
这一题也是这么做的:
多开一个维度按从大到小存储所组成的前k大值
首先按照01背包状态转移方程考虑:
-
dp[j]=max(dp[j],dp[j-a[i].v]+cost);
可以看到j是由这俩转移过来的,那么我们也可以将之变形成:
-
每一次都比较dp[j]y与dp[j-a[i].v]+cost两个体积所能装的前k大值。
这个比较的代码实现是:注意两个前K大值不能出现数字重复
for(k=1;k<=s;k++)
{
b[k]=dp[j-a[i].v][k]+a[i].c;//取后组成方式
c[k]=dp[j][k];//不取的组成方式
}
b[k]=-1,c[k]=-1;
int cnt=1,l=1,r=1;
while(cnt<=s&&(b[l]>=0||c[r]>=0))
{
if(b[l]>=c[r])
{
dp[j][cnt]=b[l++];
}
else dp[j][cnt]=c[r++];
if(dp[j][cnt]!=dp[j][cnt-1]) cnt++;//避免出现两次重复数据,只有两次数据之间不重复才往下走
}
AC代码
#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
struct ab
{
int c,v;
}a[10005];
int main()
{
int n,m,i,j,k,s,t,dp[1005][500],b[500],c[500];
scanf("%d",&t);
while(t--)
{
scanf("%d %d %d",&n,&m,&s);
for(i=1;i<=n;i++)
{
scanf("%d",&a[i].c);
}
for(i=1;i<=n;i++) scanf("%d",&a[i].v);
int cnt=0,d;
memset(dp,0,sizeof(dp));//dp[v][j]:v是以v为体积,j是第大值
memset(b,0,sizeof(b));
memset(c,0,sizeof(c));
for(i=1;i<=n;i++)//
{
for(j=m;j>=a[i].v;j--)
{
for(k=1;k<=s;k++)
{
b[k]=dp[j-a[i].v][k]+a[i].c;
c[k]=dp[j][k];
}
b[k]=-1,c[k]=-1;
int cnt=1,l=1,r=1;
while(cnt<=s&&(b[l]>=0||c[r]>=0))
{
if(b[l]>=c[r])
{
dp[j][cnt]=b[l++];
}
else dp[j][cnt]=c[r++];
if(dp[j][cnt]!=dp[j][cnt-1]) cnt++;//避免出现两次重复数据,只有两次数据之间不重复才往下走
}
}
}
/*for(i=1;i<=s;i++)
{
printf("dp[%d][%d]=%d\n",m,i,dp[m][i]);
}*/
printf("%d\n",dp[m][s]);
}
return 0;
}