1905. 混合背包

题目描述

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

物品一共有三类：

1. 第一类物品只能用 1 次（01背包）；

2. 第二类物品可以用无限次（完全背包）；

3. 第三类物品最多只能用si​ 次（多重背包）；

每种体积是 vi​，价值是 wi​。

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

输出最大价值。

输入

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

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

1. si​=−1 表示第 i 种物品只能用 1 次；

2. si​=0 表示第 i 种物品可以用无限次；

3. si​>0 表示第 i 种物品可以使用 si​ 次；

数据范围

0<N,V≤1000, 0<vi,wi≤1000, −1≤si≤1000。

输出

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

样例

输入

复制

4 5
1 2 -1
2 4 1
3 4 0
4 5 2

输出

复制

8

思路

很简单，只需要加个判断语句就行了！

代码

#include<bits/stdc++.h>
using namespace std;
int n,i,m;
int s[1001];
int main(){
	cin>>m>>n;
	for(i=1;i<=m;i++){
		int a,b,c;
		cin>>a>>b>>c;
		if(c==-1){//可以使用无限次
			for(int j=n;j>=a;j--)s[j]=max(s[j],s[j-a]+b);
		}else if(c==0){//只能使用一次
			for(int j=a;j<=n;j++)s[j]=max(s[j],s[j-a]+b);
		}else{//可以使用多次
			for(int k=0;k<c;k++)
				for(int j=n;j>=a;j--)s[j]=max(s[j],s[j-a]+b);
		}
	}
	cout<<s[n];
	return 0;
}