Leetcode 1155. 掷骰子的N种方法

Leetcode 1155. 掷骰子的N种方法

题目

这里有 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

思路

• 这道题目的思想就是DP
• 设置边界值dp[0][0] = 1
• 递推方程为dp[i][sum] = (dp[i - 1][sum - margin] % mod + dp[i][sum]) % mod

注：
i表示第几个骰子
sum表示骰子的总点数
margin表示第i个骰子掷出的点数

代码 —— golang

func numRollsToTarget(d int, f int, target int) int {
    const mod = 1e9 + 7
	dp := make([][]int, d + 1)

	for i := range dp {
		dp[i] = make([]int, target + 1)
	}

	// 边界值
	dp[0][0] = 1

	for i := 1;i <= d;i++ {
		for margin := 1;margin <= f;margin++ {
			for sum := margin;sum <= target;sum++ {
				dp[i][sum] = (dp[i - 1][sum - margin] % mod + dp[i][sum]) % mod
			}
		}
	}

	return dp[d][target]
}