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];

    }
}

 

你可能感兴趣的:(JAVA程序设计:掷骰子的N种方法(LeetCode:1155))