【LeetCode/力扣】879. 盈利计划

1 题目描述

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

---

集团里有 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) 。

---

2 代码/C++

class Solution {
public:
    int profitableSchemes(int n, int minProfit, vector<int>& group, vector<int>& profit) {
        // 建立动态规划数组，两个维度分别为人数和利润目标        
        // 优化空间的动态规划，省略一个维度，这里是项目数
        vector<vector<int>> dp(n + 1, vector<int>(minProfit + 1));
        // 初始化数组，在项目数为0、利润目标为0的情况下，不管有多少人，都只有一种计划，就是(0，0，0)
        for (int i = 0; i <= n; i++) {
            dp[i][0] = 1;
        }
        // len为项目个数，由于结果较大，所以以结果模1e9 + 7作为输出
        int len = group.size(), MOD = (int)1e9 + 7;
        // 第一层循环遍历项目数
        for (int i = 1; i <= len; i++) {
            // 读取第i个项目的人数和利润
            int members = group[i - 1], earn = profit[i - 1];
            // 第二层循环，逆序遍历，从人数n到人数members
            for (int j = n; j >= members; j--) {
                // 第三层循环，逆序遍历，从利润目标到利润0，保证第i个项目的利润都能被加进来（多多益善）
                for (int k = minProfit; k >= 0; k--) {
                    // 转移方程，一项是前i-1个项目的计划数，一项是加入第i个项目的计划数
                    // 利润项中一般的背包问题都是要求不大于利润目标，所以是 k - earn
                    // 而题目要求是不小于利润目标，所以用max(0, k - earn)，且从利润目标到利润0倒序
                    dp[j][k] = (dp[j][k] + dp[j - members][max(0, k - earn)]) % MOD;
                }
            }
        }
        return dp[n][minProfit];
    }
};

正常的背包问题，都是不能超过背包容量，所以采用逆序遍历到当前输入（当前项目所需人数），背包需要预留当然输入 earn的足够空间，即背包容量不能超过k - earn；这里的利润目标是下限，理解为至少要超过该背包容量，所以采用逆序遍历到0，如果当前输入超过背包容量可以放进去，如果当前输入小于背包容量，则需要背包中至少有超过k - earn的容量（前期项目少时够不着的方案数都是0）