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

 

class Solution {
public:
    vector<double> dicesProbability(int n) {
        vector<double> dp(6,1.0/6.0); // 这是第一轮掷骰子的结果
        for(int i=2;i<=n;i++){  // 开始计算第2轮至第n轮掷骰子的结果
            vector<double> tmp(i*5+1,0); // 初始化一个容器存储本轮的全部概率，容器大小按照5*i+1计算
            for(int j=0;j<dp.size();j++){ // 遍历上一轮的容器
                for(int k=j;k<j+6;k++){ // 上一轮容器的每个和值，都可对本轮的6个和值产生影响，位置范围为[j,j+5]
                    tmp[k] += dp[j]/6.0; // 这里除以6.0的意思是，基于上一轮的和值，在本轮投出1-6中某个点数的概率应该是：上一轮概率*(1/6)
                }
            }
            dp = tmp;
        }
        return dp;
    }
};