這裡有 d 個一樣的骰子,每個骰子上都有 f 個面,分别标号為 1, 2, ..., f。
我們約定:擲骰子的得到總點數為各骰子面朝上的數字的總和。
如果需要擲出的總點數為 target,請你計算出有多少種不同的組合情況(所有的組合情況總共有 f^d 種),模 10^9 + 7 後傳回。
示例 1:
輸入:d = 1, f = 6, target = 3
輸出:1
示例 2:
輸入:d = 2, f = 6, target = 7
輸出:6
示例 3:
輸入:d = 2, f = 5, target = 10
輸出:1
示例 4:
輸入:d = 1, f = 2, target = 3
輸出:0
示例 5:
輸入:d = 30, f = 30, target = 500
輸出:222616187
提示:
1 <= d, f <= 30
1 <= target <= 1000
思路:01背包的簡單變形,我們定義dp[i][j]表示前i個骰子朝上的點數和為j的方案數。
class Solution {
static final int mod = 1000000007;
public int numRollsToTarget(int d, int f, int target) {
long[][] dp = new long[d + 1][target + 1];
dp[0][0] = 1;
for (int i = 1; i <= d; i++)
for (int j = 1; j <= f; j++)
for (int k = target; k >= j; k--)
dp[i][k] = (dp[i][k] + dp[i - 1][k - j]) % mod;
return (int) dp[d][target];
}
}