-
Notifications
You must be signed in to change notification settings - Fork 242
/
Copy pathNumberOfDiceRollsWithTargetSum.cpp
110 lines (81 loc) · 2.75 KB
/
NumberOfDiceRollsWithTargetSum.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// https://leetcode.com/problems/number-of-dice-rolls-with-target-sum/description/
class Solution {
public:
long long modValue= 1e9+7;
// Simple Recursive Solution
// int solve(int n, int k, int target){
// if(target<0)
// return 0;
// if(n==0 && target!=0)
// return 0;
// if(n!=0 && target==0)
// return 0;
// if(n==0 && target==0)
// return 1;
// long long ans= 0;
// for(int i=1; i<=k; i++){
// ans= (ans%modValue + solve(n-1, k, target-i)%modValue)%modValue;
// }
// return (int)ans;
// }
// Recursion + Memoization (Top-Down Approach)
// int solve(int n, int k, int target, vector<vector<int>>& dp){
// if(target<0)
// return 0;
// if(n==0 && target!=0)
// return 0;
// if(n!=0 && target==0)
// return 0;
// if(n==0 && target==0)
// return 1;
// if(dp[n][target]!=-1)
// return dp[n][target];
// long long ans= 0;
// for(int i=1; i<=k; i++){
// ans= (ans%modValue + solve(n-1, k, target-i, dp)%modValue)%modValue;
// }
// dp[n][target]= (int)ans;
// return dp[n][target];
// }
// Tabulation Method (Bottom-Up Approach)
// int solve(int n, int k, int target){
// vector<vector<int>> dp(n+1, vector<int>(target+1, 0));
// dp[0][0]= 1;
// for(int dice=1; dice<=n; dice++){
// for(int t=1; t<=target; t++){
// long long ans= 0;
// for(int i=1; i<=k; i++){
// if(t-i>=0)
// ans= (ans%modValue + (dp[dice-1][t-i])%modValue)%modValue;
// }
// dp[dice][t]= (int)ans;
// }
// }
// return dp[n][target];
// }
int solve(int n, int k, int target){
vector<int> prev(target+1, 0);
vector<int> curr(target+1, 0);
prev[0]= 1;
for(int dice=1; dice<=n; dice++){
for(int t=1; t<=target; t++){
long long ans= 0;
for(int i=1; i<=k; i++){
if(t-i>=0)
ans= (ans%modValue + (prev[t-i])%modValue)%modValue;
}
curr[t]= (int)ans;
}
prev= curr;
}
return curr[target];
}
int numRollsToTarget(int n, int k, int target) {
// Simple Recursive Solution
// return solve(n, k, target);
// Recursion + Memoization (Top-Down Approach)
// vector<vector<int>> dp(n+1, vector<int>(target+1, -1));
// return solve(n, k, target, dp);
return solve(n, k, target);
}
};