leetcode之n个骰子的点数,python版本

书上写的还是比较晦涩难懂的。实际上用动态规划很好解决:

#coding=utf8
def get_ans(n):
    dp = [[0 for i in range(6*n)] for i in range(n)]

    for i in range(6):
        dp[0][i] = 1
    # print dp
    for i in range(1,n):  #1,相当于2个骰子。
        for j in range(i,6*(i+1)):   #[0,i-1]的时候,频数为0(例如2个骰子不可能投出点数和为1)
            dp[i][j] = dp[i-1][j-6] + dp[i-1][j-5] +dp[i-1][j-4]+\
                            dp[i - 1][j - 3] +dp[i-1][j-2] +dp[i-1][j-1]

    count = dp[n-1]
    return count  #算得骰子投出每一个点数的频数。再除以总的排列数即可得到频率

print get_ans(3)  #括号中的数字为骰子的个数。此代码为3个骰子时的情况。

PS:python里似乎对下标为负值的数字会默认取零。所以省去了一些麻烦。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值