【AcWing 1013. 机器分配】多重背包输出具体方案

题目链接

题意：

现在有n个分公司，m台机器，给出一个n*m的矩阵，其中第 i 行第 j 列表示第 i 个公司分到 j 台设备时的盈利，现在有m台机器，问怎样分配才能使得盈利最大。

分析：

这个题就是典型的多重背包问题，每个物品数都是m，这个数据量是10级别的，特别小，所以说朴素版的多重背包n^3完全能解决，现在的问题就是如何记录方案，在这里我开了个last数组记录i,j这个状态下取得最大值时需要拿多少台 i 设备，然后反着递推即可，下面看代码，这个题比较容易。

#include<iostream>
using namespace std;
const int N = 20;
int f[N][N];
int g[N][N],last[N][N];
int cnt[N],ans1[N],ans2[N],ans[N];
int main(){
	int n,m;
	cin>>n>>m;
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++)
			cin>>g[i][j];
	}
	for(int i=1;i<=n;i++){
		for(int j=1;j<=m;j++){
			f[i][j] = f[i-1][j];
			for(int k=1;k<=j;k++)
				if(f[i-1][j-k]+g[i][k] > f[i][j]){
					last[i][j] = k;
					f[i][j] = f[i-1][j-k]+g[i][k]; 
				}
		}
	}
	int i=n,j=m,t=0;
	while(i > 0 && j >= 0){
		if(f[i][j] == f[i-1][j]) i--;
		else{
			ans1[++t] = i;
			ans2[t] = last[i][j];
			j -= last[i][j];
			i--;
		}
	}
	cout<<f[n][m]<<endl;
	for(int i=1;i<=t;i++) ans[ans1[i]] = ans2[i];
	for(int i=1;i<=n;i++) cout<<i<<" "<<ans[i]<<endl;
	return 0;
}