求解资源分配问题 pta

某公司有3个商店A、B、C，拟将新招聘的5名员工分配给这3个商店，各商店得到新员工后，每年的赢利情况如下表所示，求分配给各商店各多少员工才能使公司的赢利最大。

输入格式:

第一行输入商店数m及员工人数n，再依次输入m+1行，每行为n+1个数，每个数（i,j）表示i商店分配j人赢利值0≤i≤m,0≤j≤n。

输出格式:

输出前m行每行两个数，分别表示商店编号及分配人数，最后一行表示公司最大赢利。

输入样例1:

3 5
0 0 0 0 0 0
0 3 7 9 12 13
0 5 10 11 11 11
0 4 6 11 12 12

输出样例1:

3 3
2 2
1 0
21

代码

 

/*
3 5
0 0 0 0 0 0
0 3 7 9 12 13
0 5 10 11 11 11
0 4 6 11 12 12
*/
#include <iostream>
using namespace std;

const int maxn = 100;
int m, n; // m为商店个数，n为员工个数
int dp[maxn][maxn];
int a[maxn][maxn];	  //分配员工数和赢利情况的表
int path[maxn][maxn]; //存储路径，记录分配结果

//设dp[i][j]表示，前i个商店分配j个员工能获得的最大利益值
void DP()
{
	//先处理边界1
	for (int i = 1; i <= m; i++)
	{
		dp[i][0] = 0;
		path[i][0] = 0;
	}
	//边界2
	for (int j = 1; j <= n; j++)
	{
		dp[1][j] = a[1][j];
		path[1][j] = j;
	}

	//再来其他情况，遍历顺序从上到下，从左到右，是常规顺序
	for (int i = 2; i <= m; i++)
	{
		for (int j = 1; j <= n; j++)
		{
			int max_res = 0;
			int max_j = 0;
			for (int k = 0; k <= j; k++) //找出最大的那个
			{
				if (dp[i - 1][k] + a[i][j - k] > max_res)
				{
					max_res = dp[i - 1][k] + a[i][j - k];
					max_j = j - k; //记录下本商店在最大利益时分配的人，注意，是j - k表示本商店的分配人数！
				}
			}
			dp[i][j] = max_res;
			path[i][j] = max_j;
		}
	}
}

int main()
{
	cin >> m >> n;
	for (int i = 0; i <= m; i++)
	{
		for (int j = 0; j <= n; j++)
		{
			cin >> a[i][j];
		}
	}

	DP();
	//输出具体方案
	cout << m << " " << path[m][n] << endl;
	int r = n - path[m][n]; //剩余的人数
	for (int i = m - 1; i >= 1; i--)
	{
		cout << i << " " << path[i][r] << endl;
		r = r - path[i][r];
	}
	cout << dp[m][n] << endl;
	return 0;
}