leetcode--n个骰子的点数(动态规划)

这两天做虚拟化的openstack实验,头秃,赶快做点题目

题目来源:链接

题目描述:

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

 

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

 

代码实现:

def func(n):
    dp = [0] * (6*n+1)
    for i in range(1,7):
        dp[i] = 1/6
    for _ in range(n-1):
        # 内层循环只能是6*n 到 1,可能出现的点数也是这样的
        for j in range(6*n,0,-1):
            dp[j] = sum(dp[max(0,j-6):j]) * 1/6
    return dp[n:]

解题思路:

首先要对随机数学的概率问题要理解

比如我们掷2个骰子:

和为7的情况有6种:{(1,6), (2,5), (3,4), (4,3), (5,2), (6,1)}

和为8的情况有5种:{(2,6), (3,5), (4,4), (5,3), (6,2)}

只需要一个数组就够了,不需要二维数组

dp[i]表示掷出i的次数

 

你可能感兴趣的:(LeetCode,动态规划)