【leetcode】盈利计划（二维费用01背包）

集团里有 n 名员工，他们可以完成各种各样的工作创造利润。

第 i 种工作会产生 profit[i] 的利润，它要求 group[i] 名成员共同参与。如果成员参与了其中一项工作，就不能参与另一项工作。

工作的任何至少产生 minProfit 利润的子集称为盈利计划。并且工作的成员总数最多为 n 。

有多少种计划可以选择？因为答案很大，所以 返回结果模 10^9 + 7 的值。

示例 1：

输入：n = 5, minProfit = 3, group = [2,2], profit = [2,3]
输出：2
解释：至少产生 3 的利润，该集团可以完成工作 0 和工作 1 ，或仅完成工作 1 。
总的来说，有两种计划。
示例 2：

输入：n = 10, minProfit = 5, group = [2,3,5], profit = [6,7,8]
输出：7
解释：至少产生 5 的利润，只要完成其中一种工作就行，所以该集团可以完成任何工作。
有 7 种可能的计划：(0)，(1)，(2)，(0,1)，(0,2)，(1,2)，以及 (0,1,2) 。

提示：

1 <= n <= 100
0 <= minProfit <= 100
1 <= group.length <= 100
1 <= group[i] <= 100
profit.length == group.length
0 <= profit[i] <= 100

链接：https://leetcode-cn.com/problems/profitable-schemes

思路分析

总共有n名员工，至少要产生minProfit利润。
每份工作消耗gourp[i]名员工，产生profit[i]的价值。
总共有group.size()份工作。
对于每份工作，我们可以选择做还是不做。最后，我们要求价值>=minProfit的方案。
看到这，应该会想到和二位费用01背包很像了吧。工作就是物品，员工就是容量，利润就是价值。

那么，我们可以定义：
dp[j][k]表示对于前i份工作，此时还拥有j名员工，还需要产生k个价值。i省略，通过滚动数组优化了空间。

dp[j][k] = dp[j][k] + dp[j-group[i]][k-profit[i]]

class Solution {
public:
    int profitableSchemes(int n, int m, vector<int>& group, vector<int>& profit) {
        int mod = 1e9+7;
        int dp[n+1][m+1];
        memset(dp,0,sizeof(dp));

        //利润为0时不管有多少成员都是一个方案
        for(int i = 0;i <= n;i++) dp[i][0] = 1;

        //遍历工作
        for(int i = 0;i < group.size();i++)
        {
            //倒序遍历成员，01背包滚动数组
            for(int j = n;j >= group[i];j--)
            {
                //遍历利润
                for(int k = 0;k <= m;k++)
                {
                    //工作产生的价值可以超过minProfit，因此用max
                    if(j >= group[i]) dp[j][k] = (dp[j-group[i]][max(k-profit[i],0)] + dp[j][k]) % mod;
                }
            }
        }
        return dp[n][m];
    }
};