這是我參與更文挑戰的第9天
,活動詳情查看更文挑戰
題目
集團里有 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
解題思路
數組含義
dp[i][j][k]表示前i份工作,j名員工,利潤至少為k的情況下,計劃的數量
狀態轉移
- j>=group[i]表示當前人員充足,可以承接這個任務,所以可以選擇是否接受該任務
dp[i+1][j][k]=(dp[i][j][k]+dp[i][j-group[i]][Math.max(0,k-profit[i])])%mod;
因為第三維代表的是利潤至少為k的情況,所以k-profit[i]可能為負數,因此需要賦值為0
- j<group[i]表示當前人員不夠,不能承接這個任務
dp[i+1][j][k]=dp[i][j][k];
代碼
class Solution {public int profitableSchemes(int n, int minProfit, int[] group, int[] profit) {int m=profit.length,mod=(int) 1e9+7;int[][][] dp = new int[m+1][n+1][minProfit+1];dp[0][0][0]=1;for (int i=0;i<m;i++){for (int j=0;j<=n;j++){for (int k=0;k<=minProfit;k++) {if(j>=group[i]){dp[i+1][j][k]=(dp[i][j][k]+dp[i][j-group[i]][Math.max(0,k-profit[i])])%mod;}else {dp[i+1][j][k]=dp[i][j][k];}}}}int res=0;for (int j=0;j<=n;j++){res=(res+dp[m][j][minProfit])%mod;}return res;}
}