파비의 매일매일 공부기록

2023.12.27 Today's Challenge 본문

Problem Solving/LeetCode

2023.12.27 Today's Challenge

fabichoi 2023. 12. 27. 23:45

https://leetcode.com/problems/number-of-dice-rolls-with-target-sum/

 

Number of Dice Rolls With Target Sum - LeetCode

Can you solve this real interview question? Number of Dice Rolls With Target Sum - 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

leetcode.com

Brute Force로 푸는 문제

class Solution:
    def numRollsToTarget(self, n: int, k: int, target: int) -> int:
        mod = 10**9 + 7
        prev = [0] * (target + 1)
        curr = [0] * (target + 1)
        
        prev[0] = 1

        for i in range(1, n+1):
            for j in range(1, target+1):
                ans = 0
                for x in range(1, k+1):
                    if j-x >= 0:
                        ans += prev[j-x] % mod
                curr[j] = ans
            prev = curr[:]
        
        return int(prev[target] % mod)
반응형

'Problem Solving > LeetCode' 카테고리의 다른 글

2023.12.29 Today's Challenge  (0) 2023.12.29
2023.12.28 Today's Challenge  (0) 2023.12.28
2023.12.26 Today's Challenge  (0) 2023.12.26
2023.12.25 Today's Challenge  (1) 2023.12.25
2023.12.24 Today's Challenge  (0) 2023.12.24
Comments