动态规划--资源分配问题

问题描述：

资源分配问题是将数量一定的一种或若干种资源(原木料、资金、设备或劳动力等)合理地分配给若干个使用者,使总收益最大。

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

 

解析：

其实就是完全背包的变形

用dp[i][j]表示为前i个商店共分配j个人时盈利的最大值，状态转移方程如下（k表示为i商店分配的人数）：

dp[i][j] = max(dp[i][j], dp[i-1][j-k] + c[i][k])

求解过程如下（括号内红字表示选的人数，画框的表示倒退求解的过程）：

具体代码：

#include <iostream>
#include <cstdio>
#include <vector>

using namespace std;
const int MAX = 30;

vector<int> ans;

int c[MAX][MAX], dp[MAX][MAX]; // dp[i][j]表示前i个车间共分配j个人
int pnum[MAX][MAX];
int n, m;

int DP_source()
{
    // 初始化
    for(int i = 0; i <= n; i++)
        dp[i][0] = 0;

    for(int i = 0; i <= m; i++)
        dp[0][i] = 0;


    int sel_num = 0; // 记录第i个车间分配j人时应分配的人数
    for(int i = 1; i <= n; i++)
    {
        for(int j = 1; j <= m; j++)
        {
            sel_num = 0;

            for(int k = 0; k <= j; k++) // 枚举当前车间分配k人时的最大值
            {
                if(dp[i][j] < dp[i - 1][j - k] + c[i][k]) // 更新答案
                {
                    dp[i][j] = dp[i - 1][j - k] + c[i][k];
                    sel_num = k;
                }
            }

            // 循环结束后找到为前i个车间共分配j人时第i个车间应分配的人数

            pnum[i][j] = sel_num;
        }
    }

    return dp[n][m];

}

void print_allocate() // 输出每个车间应分配的人数
{
    int r, s;
    s = pnum[n][m]; // 最后一个车间应该选择的人数
    r = m - s; // 剩余人数
    for(int i = n; i >= 1; i--)
    {
        printf("%d 商店分配 %d 人\n", i, s);

        s = pnum[i- 1][r];
        r -= s;
    }
}

int main()
{

    scanf("%d%d", &n, &m);
    for(int i = 1; i <= n; i++)
        for(int j = 1; j <= m; j ++)
        scanf("%d", &c[i][j]);

    printf("max = %d\n", DP_source());
    print_allocate();


    return 0;
}

/*
3 5
3 7 9 12 13
5 10 11 11 11
4 6 11 12 12
*/