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]
相关推荐
卡卡卡卡罗特14 分钟前
每日mysql
数据结构·算法
chao_7891 小时前
二分查找篇——搜索旋转排序数组【LeetCode】一次二分查找
数据结构·python·算法·leetcode·二分查找
lifallen2 小时前
Paimon 原子提交实现
java·大数据·数据结构·数据库·后端·算法
lixzest2 小时前
C++ Lambda 表达式详解
服务器·开发语言·c++·算法
EndingCoder2 小时前
搜索算法在前端的实践
前端·算法·性能优化·状态模式·搜索算法
丶小鱼丶2 小时前
链表算法之【合并两个有序链表】
java·算法·链表
不吃洋葱.2 小时前
前缀和|差分
数据结构·算法
是白可可呀4 小时前
LeetCode 169. 多数元素
leetcode
jz_ddk5 小时前
[实战]调频(FM)和调幅(AM)信号生成(完整C语言实现)
c语言·算法·信号处理
CloudAce云一5 小时前
谷歌云代理商:谷歌云TPU/GPU如何加速您的AI模型训练和推理
算法