剑指offer系列-面试题60. n个骰子的点数(python)

1. 题目

把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。

2. 解题思路

详情见 【n个骰子的点数】：详解动态规划及其优化方式

大致的思路是，将问题分解。

扔n个骰子 = 扔n-1个骰子 + 扔1个骰子

3. 代码实现

3.1

class Solution:
    def twoSum(self, n: int) -> List[float]:
        total = 6 ** n
        dp = [0] * (6 * n + 1)
        res = []
        for j in range(1, 7): # 1.扔第一个骰子的时候都是1次
            dp[j] = 1
        for i in range(2, n + 1): # 2.从扔第2个骰子开始，
            for j in range(6 * n, i - 1, -1): # 3.从后向前更新
                dp[j] = 0
                for k in range(1, 7):
                    if j - k >= i - 1:
                        dp[j] += dp[j - k]
        for j in range(n, 6 * n + 1):
            res.append(dp[j] * 1.0 / total)
        return res

4. 总结

无

5. 参考文献

[1] 剑指offer丛书
[2] 剑指Offer——名企面试官精讲典型编程题
[3] 【n个骰子的点数】：详解动态规划及其优化方式