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()));
        }
    }
}
posted @ 2021-12-24 17:13  Tianyiya  阅读(493)  评论(0)    收藏  举报