879. 盈利计划

Posted 2021-06-27 ai52learn

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

class Solution {
public:
    int profitableSchemes(int n, int minProfit, vector<int>& group, vector<int>& profit) {
        int sum = 0;
        for(int i = 0;i<profit.size();i++)
        {
            sum+=profit[i];
        }
        if(sum<minProfit) return 0;
        int d[10001][101]={0};
        d[0][0] = 1;
        int x,y;
        const int INF = 1e9+7;
        for(int i = 0; i < profit.size();i++)
        {
            for(int j = sum;j>=0;j--)
            for(int k = n;k>=0;k--)
            {
                x = j-profit[i];
                y = k-group[i];
                if(x>=0&&y>=0&&d[x][y]>0)
                {
                    d[j][k]=(d[j][k]+d[x][y])%INF;
                }
            }
        }
        int ret=0;
        for(int i = minProfit;i<=sum;i++)
        for(int j = 0;j<=n;j++)
            ret=(ret+d[i][j])%INF;
        return ret;
    }
};