【洛谷P1858】多人背包
多人背包
题目描述:
DD 和好朋友们要去爬山啦!他们一共有 \(K\) 个人,每个人都会背一个包。这些包的容量是相同的,都是 \(V\)。可以装进背包里的一共有 \(N\) 种物品,每种物品都有给定的体积和价值。在 DD 看来,合理的背包安排方案是这样的:
- 每个人背包里装的物品的总体积恰等于包的容量。
- 每个包里的每种物品最多只有一件,但两个不同的包中可以存在相同的物品。
- 任意两个人,他们包里的物品清单不能完全相同。
在满足以上要求的前提下,所有包里的所有物品的总价值最大是多少呢?
输入格式:
第一行有三个整数:\(K\)、\(V\)、\(N\)。
第二行开始的 \(N\) 行,每行有两个整数,分别代表这件物品的体积和价值。
输出格式:
只需输出一个整数,即在满足以上要求的前提下所有物品的总价值的最大值。
样例输入:
2 10 5 3 12 7 20 2 4 5 6 1 1
样例输出:
57
数据范围:
总人数 \(K<=50\)。
每个背包的容量 \(V<=5000\)。
物品种类数 \(N<=200\)。
其它正整数都不超过 5000。
输入数据保证存在满足要求的方案。
时间限制:
1S
空间限制:
128M
提示:
remove!!!
题解
多人背包?那不就是背包问题吗?
看题目好像还是01背包。
有\(K\)个人,每个人的背包装的东西不能相同,那么不就是前\(K\)大01背包吗?
我们在原来的01背包上增加一维来维护前\(K\)大就行了。
\(dp[i][j]\)表示体积为\(i\)时第\(j\)大的价值是多少。
转移的时候要注意一下,要用类似于归并排序合并时的方法转移。(具体如何实现看代码)
int u1=1,u2=1;
while(u1+u2-1<=k && (dp[j+a][u1]!=-1 || dp[j][u2]!=-1)){
if(dp[j+a][u1]!=-1 && (dp[j][u2]==-1 || dp[j+a][u1]>dp[j][u2]+b)){us[u1+u2-1]=dp[j+a][u1];u1++;}
else{us[u1+u2-1]=dp[j][u2]+b;u2++;}
}
for(int o=1;o<=u1+u2-2;o++)
dp[j+a][o]=us[o];//更新dp中的值
因为只求前\(k\)大的和,所以\(dp\)过程中只用维护前\(k\)大就行了。
注意最后更新时循环是1到\(u1+u2-2\),因为前面循环结束时\(u1\)或者\(u2\)会多加一次。
要注意这里不能用1到\(k\),因为前\(k\)个可能不一定全部被更新过,如果\(dp[j+a][k]\)没有被更新过,就用它去更新下一个,就会出bug。
上代码:
#include<bits/stdc++.h>
using namespace std;
int k,v,n;
int a,b;
int dp[5009][59];
int us[59];
int ans;
int main(){
scanf("%d%d%d",&k,&v,&n);
memset(dp,-1,sizeof(dp));
dp[0][1]=0;
for(int i=1;i<=n;i++){
scanf("%d%d",&a,&b);
for(int j=v-a;j>=0;j--){
if(dp[j][1]!=-1){//没有这个判断对于结果没有影响,只是用来节省一点点时间
int u1=1,u2=1;
while(u1+u2-1<=k && (dp[j+a][u1]!=-1 || dp[j][u2]!=-1)){
if(dp[j+a][u1]!=-1 && (dp[j][u2]==-1 || dp[j+a][u1]>dp[j][u2]+b)){
us[u1+u2-1]=dp[j+a][u1];u1++;
}else{us[u1+u2-1]=dp[j][u2]+b;u2++;}
}
for(int o=1;o<=u1+u2-2;o++)
dp[j+a][o]=us[o];
}
}
}
for(int j=1;j<=k;j++)
ans+=dp[v][j];
printf("%d",ans);
return 0;
}