面试题60. n个骰子的点数-Python题解

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

你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

示例 1:

输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]
示例 2:

输入: 2
输出: [0.02778,0.05556,0.08333,0.11111,0.13889,0.16667,0.13889,0.11111,0.08333,0.05556,0.02778]

限制：

1 <= n <= 11

class Solution:
    def dicesProbability(self, n: int) -> List[float]:

        dp = [[ 0 for _ in range(6*n+1) ] for _ in range(n +  1)]
        # 给第一个骰子值进行赋值,每一个点都出现一次的次数相同,则都为1
        for i in range(1,7):
            # 第一个骰子1面出现的次数为1,则就把i=1这个位置设置成1
            dp[1][i] = 1
        # print(dp)
        # 通过动态规划得到后面骰子出现的次数
        for i in range(2,n + 1):
            for j in range(i, i * 6 + 1):
                # j 代表n个骰子和可能出现的数字
                for k in range(1,7):
                    # k表示第n个骰子一共有几个表
                    if j > k:
                        #表示的前i-1个骰子的j-k的值加上第i个骰子的k值,最终等于j这个值,我们得到当前骰子的值加上前i-1个骰子的和的哪些等于前i个骰子可能出现的和
                        # print(j, j - k)
                        dp[i][j] += dp[i - 1][j - k]
        res = []
        # print(dp)
        # n个骰子的最小值为n,最大值为n*6
        for i in range(n, n*6 + 1):
            res.append(dp[n][i] * 1.0 / 6**n)
        return res