参考力扣879 盈利计划的题解,我认为总结的很好,于是将题解里面的c语言代码改写成python,学习一下,变成自己的东西。

从力扣879说起

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

这种题一打眼就感觉能用DFS/回溯做,但是这种暴力解法通常会超时,果然师弟笔试题只通过了50%。

当时写的代码主要函数是这样的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
cost = [2,2]
profit = [2,3]
count = 0
def dfs(idx,cur_p,m):
nonlocal count
if m<0:
return
if idx==len(cost):
if cur_p>=p:
count+=1
return
dfs(idx+1,cur_p+profit[idx],m-cost[idx])
dfs(idx+1,cur_p,m)
dfs(0,0,5)

这种想法就是不断递归,在每一个任务做出选择,做或不做,当遍历了所有任务或者是工人不够用了的时候,跳出递归。

后来得知,这是力扣原题hard,就在879下面看到了一篇不错的题解。

01背包,好好学学。

01背包类题目分为最优解最优解和两种类型。

最优解

最优解一般模板为dp[i][j]=max(dp[i-1][j],dp[i-1][j-w[i]]+v[i])
空间压缩后为dp[j]=max(dp[j],dp[j-w[i]]+v[i])
v[i]与边界的设置,根据题目要求进行更改。