60. n个骰子的点数

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

思路：动态规划

假设n-1个骰子的解为f(n-1)，此时添加一枚骰子，n个骰子的点数和为x的概率f(n,x)为

$f(n,x)=\sum _{i=1}^{6}f(n-1,x-i)\times \frac{1}{6}$

如果按照此公式计算，会出现数组越界问题。

因此，考虑由f(n-1,x)推导f(n-1,x+i)

class Solution {
public:
    vector<double> dicesProbability(int n) {
        vector<double> dp(6,1.0/6.0);
        for(int i=2;i<=n;++i){
            vector<double> tmp(5*i+1,0);
            for(int j=0;j<dp.size();++j){
                for(int k=0;k<6;++k){
                    tmp[j+k]+=dp[j]/6.0;
                }
            }
            dp=tmp;
        }
        return dp;
    }
};

时间复杂度 O($n^2$) 计算的循环次数为$6\times 6,11\times 6,[5(n-1)+1]\times 6$总体时间复杂度为$O((6+5(n-1)+1)\times (n-1)/2)$，即O($n^2$)

空间复杂度 O(n) 需要tmp辅助数组