一句话题意已经在题名里了。
显然,这里必须要决策的过程当中不断地记录下前k优解。我们在0-1背包的基础上增加一维f[k][j]表示背包装的物品体积为j时的第k优解。
对于每次决策f[j]=max(f[j],f[j-w[i]]+v[i]) 我们都让f[1~k][j]宇f[1~k][j-w[i]+v[i]合并取前k优解。显然,如果f[1~k]是乱序的,每次搜索一遍的 时间复杂度为O(k^2) 即总时间复杂度为O(n*V*k^2) 接近10^9,炸了。
所以我们尝试让f[1~k]有序,那么每次合并时只需采取类似归并排序中合并的操作,在两个数组中逐个比较,时间复杂度O(k)。
没别的了,见code。
void init()
{
read(k);read(V);read(n);
for(int i=1;i<=n;++i) read(w[i]),read(v[i]);
}
void work()
{
for(int i=0;i<=k;++i)
for(int j=0;j<=V;++j) f[i][j]=-21472147;
f[1][0]=0;
for(int i=1;i<=n;++i)
for(int j=V;j>=w[i];--j)
{
int tott=0;
int pi=1,pj=1;
while(tott<=k)
{
if(f[pi][j]>f[pj][j-w[i]]+v[i]) ans[++tott]=f[pi++][j];
else ans[++tott]=f[pj++][j-w[i]]+v[i];
}
for(int l=1;l<=k;++l) f[l][j]=ans[l];
}
for(int i=1;i<=k;++i) Ans+=f[i][V];
printf("%d",Ans);
}