剑指-面试题60 n个骰子的点数

n个骰子的点数

题目
把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

思路 动态规划

参考leetcode题解

---

C++

class Solution {
public:
    vector<double> twoSum(int n) {
        int dp[70];
        memset(dp, 0 , sizeof(dp));
        for(int i=1; i<=6; ++i)                     //边界处理
            dp[i]=1;
        for(int i=2; i<=n; ++i)
            for(int j=6*i; j>=i; --j)
            {
                dp[j]=0;
                for(int cur=1; cur<=6; ++cur)
                {
                    if(j-cur<i-1)
                        break;
                    dp[j] += dp[j-cur];
                }
            }
        int sum = pow(6, n);
        vector<double> res;
        for(int i=n; i<=6*n; i++)
            res.push_back(dp[i] * 1.0/sum);
        return res;
    }
};

python

class Solution:
    def twoSum(self, n: int) -> List[float]:
        if n == 0:
            return []
        # 初始化 1 - 6 是 1次，7 - n 是 0 次。
        # 编号从1开始，这样方便写代码。  为了从1开始，我们只需要在数组前面随便push一个元素即可，比如本例的0
        cnts = [0] + [1] * 6 + [0] * (6 * n - 6)
        # 模拟投掷 n - 1 次
        for _ in range(n - 1):
            # 从后向前更新
            for i in range(6 * n, 0, -1):
                cnts[i] = sum(cnts[max(i - 6, 0): i])

        return filter(lambda a: a != 0, list(map(lambda a: a / 6 ** n, cnts)))