leetcode - 1155. Number of Dice Rolls With Target Sum

Description

You have n dice, and each die has k faces numbered from 1 to k.

Given three integers n, k, and target, return the number of possible ways (out of the kn total ways) to roll the dice, so the sum of the face-up numbers equals target. Since the answer may be too large, return it modulo 10^9 + 7.

Example 1:

复制代码
Input: n = 1, k = 6, target = 3
Output: 1
Explanation: You throw one die with 6 faces.
There is only one way to get a sum of 3.

Example 2:

复制代码
Input: n = 2, k = 6, target = 7
Output: 6
Explanation: You throw two dice, each with 6 faces.
There are 6 ways to get a sum of 7: 1+6, 2+5, 3+4, 4+3, 5+2, 6+1.

Example 3:

复制代码
Input: n = 30, k = 30, target = 500
Output: 222616187
Explanation: The answer must be returned modulo 109 + 7.

Constraints:

复制代码
1 <= n, k <= 30
1 <= target <= 1000

Solution

Recursive + memorization

The dp transformation equation is:
d p [ n ] [ t a r g e t ] = ∑ i = 1 k d p [ n − 1 ] [ t a r g e t − i ] dp[n][target] = \sum_{i=1}^{k}dp[n-1][target-i] dp[n][target]=i=1∑kdp[n−1][target−i]

So we could do recursive and memorization or dp.

Time complexity: o ( n ∗ t a r g e t ∗ k ) o(n*target*k) o(n∗target∗k)

Space complexity: o ( n ∗ t a r g e t ) o(n*target) o(n∗target)

DP

Feels top-bottom (recursive + memo) is easier to implement than bottom-top (dp here)

Code

Recursive + memorization

python3 复制代码
class Solution:
    def numRollsToTarget(self, n: int, k: int, target: int) -> int:
        def helper(n: int, target: int) -> int:
            if (n, target) in memo:
                return memo[(n, target)]
            if n == 1:
                if 0 < target <= k:
                    memo[(n, target)] = 1
                else:
                    memo[(n, target)] = 0
                return memo[(n, target)]
            res = 0
            for i in range(1, k + 1):
                res += helper(n - 1, target - i)
                res %= mod_val
            memo[(n, target)] = res
            return memo[(n, target)]
        mod_val = 1000000007
        memo = {}
        return helper(n, target)

DP

python3 复制代码
class Solution:
    def numRollsToTarget(self, n: int, k: int, target: int) -> int:
        mod_val = 1000000007
        dp = [[0] * (target + 1) for _ in range(n + 1)]
        # init
        for pseudo_target in range(1, min(target + 1, k + 1)):
            dp[1][pseudo_target] = 1
        for pseudo_n in range(2, n + 1):
            for pseudo_t in range(1, target + 1):
                for i in range(1, k + 1):
                    if pseudo_t - i >= 0:
                        dp[pseudo_n][pseudo_t] += dp[pseudo_n - 1][pseudo_t - i]
                dp[pseudo_n][pseudo_t] %= mod_val
        return dp[-1][-1]
相关推荐
弥彦_10 分钟前
cf1925B&C
数据结构·算法
YuTaoShao31 分钟前
【LeetCode 热题 100】994. 腐烂的橘子——BFS
java·linux·算法·leetcode·宽度优先
Wendy14418 小时前
【线性回归(最小二乘法MSE)】——机器学习
算法·机器学习·线性回归
拾光拾趣录8 小时前
括号生成算法
前端·算法
渣呵9 小时前
求不重叠区间总和最大值
算法
浮生带你学Java9 小时前
2025Java面试题及答案整理( 2025年 7 月最新版,持续更新)
java·开发语言·数据库·面试·职场和发展
拾光拾趣录9 小时前
链表合并:双指针与递归
前端·javascript·算法
好易学·数据结构9 小时前
可视化图解算法56:岛屿数量
数据结构·算法·leetcode·力扣·回溯·牛客网
香蕉可乐荷包蛋10 小时前
AI算法之图像识别与分类
人工智能·学习·算法
chuxinweihui11 小时前
stack,queue,priority_queue的模拟实现及常用接口
算法