面试题60. n个骰子的点数（动态规划）

把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

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

思路：二维动态规划

def twoSum(n):
    res = []
    # 要求每一个出现的点数出现的总数，开辟一个二维数组dp dp[n][j]表示n个骰子投掷后，总数为j的出现次数
    dp = [[0 for _ in range(6 * n + 1)] for _ in range(n + 1)]
    # (1) 初始状态
    for j in range(1, 7):  # 初始化第一行
        dp[1][j] = 1
    for i in range(2, n + 1):  # 每一行
        # 每一列，要计算的每一列的数量是不一样的，为6i,即i枚骰子，有6i种结果，需要填充的数量为6i
        for j in range(i, 6 * i + 1):
            for num in range(1, 7):
                if j - num >= 1 and dp[i - 1][j - num] > 0:
                    dp[i][j] += dp[i - 1][j - num]
    for i in range(6 * n + 1):
        if dp[n][i] / 6 ** n > 0:
            res.append(dp[n][i] / 6 ** n)
    return res


print(twoSum(20))

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof