题目:
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
思路:
参考大神解法:https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof/solution/nge-tou-zi-de-dian-shu-dong-tai-gui-hua-ji-qi-yo-3/
dp[i]:投掷完当前轮后,点数和为i时,可能的情况数(i<投掷轮数时,dp[i]为无效值)
每一轮更新时,因为要使用到上一轮的结果。故应从大到小更新,否则会出现差错。
解答:
class Solution:
def twoSum(self, n: int) -> List[float]:
#为方便理解,索引0不取,后面取到最大索引6*n
dp = [0 for _ in range(6*n+1)]
#初始化dp,第一轮的抛掷
for i in range(1,7):
dp[i]=1
#从第二轮抛掷开始算
for i in range(2,n+1):
#第二轮抛掷最小和为2,从大到小更新对应
for j in range(6*n,i-1,-1):
#每次投掷要从0更新dp[j]大小,点数和出现的次数要重新计算
dp[j]=0
#每次抛掷的点数
for cur in range(1,7):
#意思为上一轮的最小点数和为i-1
if j -cur < i - 1:
break
#根据上一轮来更新当前轮数据
dp[j]+=dp[j-cur]
#一共抛掷了6^n次
sum_ = 6**n
res = []
for i in range(n,6*n+1):
res.append(dp[i]*1.0/sum_)
return res