剑指Offer-60 n个骰子的点数

最新推荐文章于 2021-02-07 10:36:12 发布

独行小猪

最新推荐文章于 2021-02-07 10:36:12 发布

阅读量286

点赞数

本文链接：https://blog.csdn.net/u013796880/article/details/84931662

版权

题目：

将一个骰子投掷n次，获得的总点数为s，s的可能范围为n~6n。掷出某一点数，可能有多种掷法，例如投掷2次，掷出3点，共有[1,2],[2,1]两种掷法。
请求出投掷n次，掷出n~6n点分别有多少种掷法。
样例1
输入：n=1
输出：[1, 1, 1, 1, 1, 1]
解释：投掷1次，可能出现的点数为1-6，共计6种。每种点数都只有1种掷法。所以输出[1, 1, 1, 1, 1, 1]。
样例2
输入：n=2
输出：[1, 2, 3, 4, 5, 6, 5, 4, 3, 2, 1]
解释：投掷2次，可能出现的点数为2-12，共计11种。每种点数可能掷法数目分别为1,2,3,4,5,6,5,4,3,2,1。
所以输出[1, 2, 3, 4, 5, 6, 5, 4, 3, 2, 1]。

解答：

class Solution(object):
    def numberOfDice(self, n):
        """
        :type n: int
        :rtype: List[int]
        """
        dp = [[0 for i in range(6*n)] for j in range(n)]
        for i in range(6):
            dp[0][i] = 1
        for i in range(1, n):
            for j in range(i, 6*n):
                sum = 0
                for k in range(1, 7):
                    if j - k >= 0:
                        sum += dp[i - 1][j - k]
                dp[i][j] = sum
        return dp[-1][n - 1:]