1155. 掷骰子的N种方法
这里有 d 个一样的骰子,每个骰子上都有 f 个面,分别标号为 1, 2, ..., f。
我们约定:掷骰子的得到总点数为各骰子面朝上的数字的总和。
如果需要掷出的总点数为 target,请你计算出有多少种不同的组合情况(所有的组合情况总共有 f^d 种),模 10^9 + 7 后返回。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/number-of-dice-rolls-with-target-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
import java.util.Scanner;
class Solution {
private int mod = 1000000007;
public int numRollsToTarget(int n, int k, int target) {
int[] dp = new int[target + 1];
dp[0] = 1;
for (int i = 1; i <= n; ++i) {
int[] tmp = new int[target + 1];
for (int j = 1; j <= k; ++j) {
for (int s = i - 1; s + j <= target; ++s) {
tmp[s + j] = (tmp[s + j] + dp[s]) % mod;
}
}
dp = tmp;
}
return dp[target];
}
public static void main(String[] args) {
Scanner in = new Scanner(System.in);
while (in.hasNext()) {
System.out.println(new Solution().numRollsToTarget(in.nextInt(), in.nextInt(), in.nextInt()));
}
}
}
心之所向,素履以往 生如逆旅,一苇以航