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里似乎对下标为负值的数字会默认取零。所以省去了一些麻烦。

你可能感兴趣的:(刷题)