879. 盈利计划
帮派里有 G 名成员,他们可能犯下各种各样的罪行。
第 i 种犯罪会产生 profit[i] 的利润,它要求 group[i] 名成员共同参与。
让我们把这些犯罪的任何子集称为盈利计划,该计划至少产生 P 的利润。
有多少种方案可以选择?因为答案很大,所以返回它模 10^9 + 7 的值。
示例 1:
输入:G = 5, P = 3, group = [2,2], profit = [2,3]
输出:2
解释:
至少产生 3 的利润,该帮派可以犯下罪 0 和罪 1 ,或仅犯下罪 1 。
总的来说,有两种方案。
示例 2:
输入:G = 10, P = 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 <= G <= 100
0 <= P <= 100
1 <= group[i] <= 100
0 <= profit[i] <= 100
1 <= group.length = profit.length <= 100
思路
这道题相当于二维背包问题,有两种价值的东西需要考虑,这道题需要考虑的就是人数和利润,使用dp去求解,详细思路看c++代码!
class Solution {
public:
int profitableSchemes(int g, int p, vector<int>& grp, vector<int>& prf) {
int M = 1000000007;
vector<vector<int> > result(p + 1, vector<int>(g + 1));
//因为索引从0开始,加1的目的是让数组显示赚p利润,有g个人
w[0][0] = 1;//赚钱为0,0个人只有一种结果
for (int i = 0; i < grp.size(); ++i)
{//有几组
int ng = grp[i], pr = prf[i];//对应的人数和利润
for (int j = p; j >= 0; --j)
{//利润
for (int k = g; k >= 0; --k)
{//人数
if (k + ng > g|!w[j][k])
//大于总人数或者方法为空就忽略
continue;
(result[min(p, j + pr)][k + ng] += result[j][k]) %= M;
//min(p,j+pr)意味着如果j+pr利润大于p,都把利润大的事件加在利润为p上,而且数组result不用开很大去适应利润大于p的情况
//因为测试例子中给的都是选(0),(1),(2),(0,1),(0,2),(1,2),以及 (0,1,2)这些情况
//如果直接用dp,给result的一维和二维不断加人数和利润,这就相当于把这些情况加到同一个集合里面,最后直接输出结果就好
}
}
}
int ret = 0;
for (int i = 0; i <= g; ++i)
(ret += w[p][i]) %= M;
return ret;
}
};
注意
本题之所以两个for循环都是使用从p和g开始的,是因为如果从小到大的话,就会出现result中一维和二维都有减的情况,会出现越界,需要麻烦的去判断越界问题