1. 程式人生 > 其它 >1155. 擲骰子的N種方法

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()));
        }
    }
}
心之所向,素履以往 生如逆旅,一葦以航