1. 题目
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
2. 解题思路
大致的思路是,将问题分解。
扔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个骰子的点数】:详解动态规划及其优化方式