洛谷动态规划训练（4）：分组背包 P1757 通天之分组背包

P1757 通天之分组背包

题目背景
直达通天路·小A历险记第二篇

题目描述
自01背包问世之后，小A对此深感兴趣。一天，小A去远游，却发现他的背包不同于01背包，他的物品大致可分为k组，每组中的物品相互冲突，现在，他想知道最大的利用价值是多少。

输入格式
两个数m,n，表示一共有n件物品，总重量为m

接下来n行，每行3个数ai,bi,ci，表示物品的重量，利用价值，所属组数

输出格式
一个数，最大的利用价值

输入输出样例
输入 #1复制
45 3
10 10 1
10 5 1
50 400 2
输出 #1复制
10
说明/提示
1<=m<=1000 1<=n<=1000 组数t<=100

问题分析

这个问题是一个分组背包的问题。特点是所有物品分成k组，对于每个组，组内的物件互相排斥 ，也就是说每个组组内选了一件物品，就不能再选同一个组内的其他物品！
对于这个问题，状态方程为
dp[k][v]=max(dp[k-1][v],dp[k-1][v-list[k][i].w]+list[k][i].v|i对应于k组内的第i个物品);
那么我们需要思考的是：
1.分类的状况是否完备。应该是完备的，因为根据是否包含k组的进行分类
2.分类的情况是否重复，是不重复的，因为如果选了第k组的物品，那么就用总钱数减去当前第i件的钱数，剩余的配置和k-1组的类似。另外，不同钱数之间也不会互相重叠，因此这个问题是完备的。

代码

#include<iostream>
#include<vector>
#include<algorithm>
#include<cstring>
using namespace std;
struct obj {
	int weight;
	int val;
	int num;
	obj(int a, int b, int c) :weight(a), val(b), num(c) {}
	obj():weight(0),val(0),num(0){}
};
int main() {
	int m, n, cn;
	cin >> m >> n;//weight,num of items
	int a, b, c;
	vector<vector<obj>> list(101,vector<obj>(1));
	int groups = 0;
	int numofparts[1001];
	for (int i = 1; i <= n; i++) {
		cin >> a >> b >> c;
		groups = max(c, groups);
		list[c].push_back(obj(a, b, c));
	}

	int dp[1001];
	memset(dp, 0, sizeof(dp));
	for (int k = 1; k <= groups; k++) {
		for (int j = m; j >= 0;j--) {
			for (int i = 1; i <= list[k].size() - 1; i++) {
				if (j >= list[k][i].val)
					dp[j] = max(dp[j], dp[j - list[k][i].weight] + list[k][i].val);
			}
		}
	}
	cout << dp[m];

	/*以下是我自己想的一种把全部表列出来然后求结果的想法，这样虽然清晰但是占用的空间比较大*/
	//int res[101] = {};
	//int dp[1001] = {};
	//for (int k = 1; k <= groups; k++) {
	//	for (int i = 1; i <= list[k].size() - 1; i++) {
	//		for (int j = m; j >= 0; j--) {
	//			if (j >= list[k][i].weight)
	//				dp[j] = max(dp[j], dp[j - list[k][i - 1].weight] + list[k][i].val);
	//		}
	//	}
	//	res[k] = dp[m];
	//	memset(dp, 0, sizeof(dp));
	//}
	//int finalres = -1;
	//for (int i = 1; i <= groups; i++) {
	//	finalres = max(finalres, res[i]);
	//}
	//cout << finalres;
	return 0;
}