天天看點

java疊骰子_JAVA程式設計:擲骰子的N種方法(LeetCode:1155)

這裡有 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];

}

}