P1757 通天之分组背包 题解

思路

#include<stdio.h>
#include<iostream>
#include<algorithm>
using namespace std;
int n,m;
int pos[1005][1005];
int dp[1005];
int w[1005],v[1005],d[1005];
int num=0;
int main(){
	/*本题其实是01背包模板题目，只是在原有的基础上，多加了条件，就是有同组的物品冲突，
	也就是同组的物品不能放在一起，所以需要用数组记录同种物品*/ 
	scanf("%d %d",&m,&n);
	for(int i=1;i<=n;i++){
		int t;
		scanf("%d %d %d",&w[i],&v[i],&t);
		//d[t]用来记录t组物品的数量 
		d[t]++;
		//num记录一共有多少组 
		num=max(num,t);
		//记录t组里面第d[t]个物品的下标，因为需要访问w和v数组 
		pos[t][d[t]]=i;
	}
	/*
	下面这段代码为易范错误，这样虽然对物品进行分组处理了，
	但是同组的物品任然会被加入到背包 
	for(int i=1;i<=num;i++){
		for(int j=1;j<=d[i];j++){
			int p=pos[i][j];
			for(int k=m;k>=w[p];k--){
				dp[k]=max(dp[k],dp[k-w[p]]+v[p]);
			}
		}
	}*/
	
	//一下为正确代码，先枚举重量，然后在枚举同组的物品，dp[j]记录最大价值
	for(int i=1;i<=num;i++){
		for(int j=m;j>=0;j--){
			for(int k=1;k<=d[i];k++){
				int p=pos[i][k];
				if(j>=w[p]){
					dp[j]=max(dp[j],dp[j-w[p]]+v[p]);
				}
			}
		}
	}
	
	printf("%d\n",dp[m]);
	
	return 0;
}