【刷题】动态规划——背包问题：机器分配【分组背包，求具体方案】

题目链接

设输入矩阵是w[N][M]
解释一下样例，第一台公司分配1台机器，获得盈利为w[1][1]=30，同理第二个公司盈利w[2][1]=20，第三个公司盈利w[3][1]=20，总共盈利70。

这题可以将每个公司看成一个组，每个组有M个物品，价值分别是w[i][1], w[i][2], …, w[i][m]，物品的体积就是分配机器的台数。如样例中公司1有3个物品，价值分别是30 40 50，体积分别是1 2 3。

那么问题就转化为：从这些组中，每组至多挑一个物品（每个公司要么分配1台、要么分配2台…），所能达到的价值和最大是多少。显然是普通分组背包问题。

将分组背包方案输出即可，需要按字典序大小，注意这边的字典序和普通背包问题不太相同，普通背包不取的方案不会输出，而这边不取的方案需要输出 i 0，它的字典序相比取更小，这意味着在取和不取得到价值相同时，不取更好。
故判断f[i][j] == f[i - 1][j - v[i][k]] + w[i][k]时，需要将k=0考虑进去。

此外，这组选了一个输出后要立即break。

#include <iostream>
using namespace std;
int n, m, f[15][20], v[15][20], w[15][20];
int ans[15];
int main() {
	scanf("%d%d", &n, &m);
	for (int i = n; i >= 1; i--) {
		for (int j = 1; j <= m; j++) {
			scanf("%d", &w[i][j]);
			v[i][j] = j;
		}
	}
	for (int i = 1; i <= n; i++) {
		for (int j = 0; j <= m; j++) {
			f[i][j] = f[i - 1][j];
			for (int k = 1; k <= j; k++) {
				if (j - v[i][k] >= 0) {
					f[i][j] = max(f[i][j], f[i - 1][j - v[i][k]] + w[i][k]);
				}
			}
		}
	} 
	printf("%d\n", f[n][m]);
	int j = m;
	for (int i = n; i >= 1; i--) {
		for (int k = 0; k <= j; k++) {
			if (j >= v[i][k] && f[i][j] == f[i - 1][j - v[i][k]] + w[i][k]) {
				ans[n - i + 1] = k;
				j -= v[i][k];
				break;
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		printf("%d %d\n", i, ans[i]);
	}
	return 0; 
}

事实上v[i][j] = j，不需要记录体积v。

#include <iostream>
using namespace std;
int n, m, f[15][20], w[15][20];
int ans[15];
int main() {
	scanf("%d%d", &n, &m);
	for (int i = n; i >= 1; i--) {
		for (int j = 1; j <= m; j++) {
			scanf("%d", &w[i][j]);
		}
	}
	for (int i = 1; i <= n; i++) {
		for (int j = 0; j <= m; j++) {
			for (int k = 0; k <= j; k++) {
				f[i][j] = max(f[i][j], f[i - 1][j - k] + w[i][k]);
			}
		}
	} 
	printf("%d\n", f[n][m]);
	int j = m;
	for (int i = n; i >= 1; i--) {
		for (int k = 0; k <= j; k++) {
			if (f[i][j] == f[i - 1][j - k] + w[i][k]) {
				ans[n - i + 1] = k;
				j -= k;
				break;
			}
		}
	}
	for (int i = 1; i <= n; i++) {
		printf("%d %d\n", i, ans[i]);
	}
	return 0; 
}