动态规划-多重背包

题意

有 N 种物品和一个容量是 V 的背包。

第 i 种物品最多有 si 件，每件体积是 vi，价值是 wi。

求解将哪些物品装入背包，可使物品体积总和不超过背包容量，且价值总和最大。
输出最大价值。

输入格式

第一行两个整数，N，V，用空格隔开，分别表示物品种数和背包容积。

接下来有 N 行，每行三个整数 vi,wi,si，用空格隔开，分别表示第 i 种物品的体积、价值和数量。

输出格式

输出一个整数，表示最大价值。

数据范围

0<N,V≤100
0<vi,wi,si≤100

输入样例

4 5
1 2 3
2 4 1
3 4 3
4 5 2

输出样例：

10

分析 （与完全背包类似，对所选个数k多了层限制s[i],即k<=s[i]；）

 

 代码

#include<bits/stdc++.h>
using namespace std;
const int N=110;

int w[N],v[N],s[N],dp[N][N];

int main(){
	std::ios::sync_with_stdio(false),std::cin.tie(nullptr);
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		cin>>v[i]>>w[i]>>s[i];
	}
	for (int i = 1; i <= n; i ++ ){
		for (int j=0; j<=m; j++){
			for(int k=0;k<=s[i]&&k*v[i]<=j;k++){
				dp[i][j]=max(dp[i][j],dp[i-1][j-k*v[i]]+k*w[i]);
			}
		}
	}
	cout<<dp[n][m]<<endl;
	
	return 0;
}

时间优化

思路：将s[i]用二进制优化分组，分为log(s[i])+x(x取决于s[i]是否剩余)组，将每组看作一个大物品，每个大物品只能取一次，能拼凑出0~s[i]的所有组合。最后将问题转变为01背包问题(对每个大物品取还是不取)。

代码

#include<bits/stdc++.h>
using namespace std;
const int N=12100;

int w[N],v[N],dp[N];

int main(){
	std::ios::sync_with_stdio(false),std::cin.tie(nullptr);
	int n,m,cnt=0;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		int a,b,s;
		cin>>a>>b>>s;
		int k=1;
		while(k<s){
			cnt++;
			v[cnt]=k*a;
			w[cnt]=k*b;
			s-=k;
			k*=2;
		}
		if(s>0){
			cnt++;
			v[cnt]=s*a;
			w[cnt]=s*b;
		}
	}
	n=cnt;
	for (int i = 1; i <= n; i ++ ){
		for (int j=m; j>=v[i]; j--){
			dp[j]=max(dp[j],dp[j-v[i]]+w[i]);
		}
	}
	cout<<dp[m]<<endl;
	
	return 0;
}