洛谷P1757 通天之分组背包

题目传送门

思路

分组背包。

01背包模板

for(int i=1;i<=n;i++){
		for(int j=T;j>=w[i];j--)dp[j]=max(dp[j],dp[j-w[i]]+v[i]);
	}

 我们需要再i和j中间加一层循环，来枚举每组取第几个的情况，状态转移方程也需要变化，由于原来的01背包时和现在的答案比较，所以即使我们加了一层循环，它也是和取前面的数的答案和现在的答案进行比较，也有可能去了前面的数。

解决方法

把dp数组增加一维，从 1到组的数量，状态转移方程变成了

if(j>=w[i][k-1]){
	dp[i][j]=max(dp[i][j],dp[i-1][j-w[i][k-1]]+v[i][k-1]);		
}
else dp[i][j]=max(dp[i][j],dp[i-1][j]);

取前面的最大值和再上一层取这个数的最大值。

代码

#include<bits/stdc++.h>
#define endl '\n';
using namespace std;
long long n,m,zu[1100],dp[1100][1100],cnt;
vector<long long> w[1100],v[1100];
int main(){
	ios::sync_with_stdio(false);
	cin.tie(0);
	cout.tie(0);
	cin>>m>>n;
	long long a,b,c;
	for(int i=1;i<=n;i++){
		cin>>a>>b>>c;
		cnt=max(cnt,c);
		zu[c]++;
		w[c].push_back(a);
		v[c].push_back(b);
	}for(int i=1;i<=cnt;i++){
		for(int k=1;k<=zu[i];k++){
			for(int j=m;j>=0;j--){
				if(j>=w[i][k-1]){
					dp[i][j]=max(dp[i][j],dp[i-1][j-w[i][k-1]]+v[i][k-1]);		
				}
				else dp[i][j]=max(dp[i][j],dp[i-1][j]);
					//cerr<<i<<" "<<j<<" "<<dp[i][j]<<endl;
					
				
			}
		}
	}cout<<dp[cnt][m];
	return 0;
}