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

剑指 Offer 60. n个骰子的点数 - 力扣（LeetCode）

概率是出现次数/总次数，n 枚骰子掷出的点数的范围是 [n, 6*n]。

dp:

dp[i][j] ，表示投掷完 i 枚骰子后，点数 j 的出现次数

dp[i][j] += dp[i-1][j-x]，x是对骰子点数进行枚举（1,2,3,4,5,6）就行了

class Solution {
public:
    vector<double> twoSum(int n) {
        //dp[i][j] ，表示投掷完 i 枚骰子后，点数 j 的出现次数
        vector<vector<int>> dp(15, vector<int>(70, 0));
        for(int i = 1; i <= 6; ++i) dp[1][i] = 1;//初始化第一个阶段（仅投掷第一枚骰子）

        for(int i = 2; i <= n; ++i){//枚举骰子的个数
            for(int j = i; j <= i * 6; ++j){//枚举可能的数值
                for(int k = 1; k <= 6; ++k){//枚举骰子的点数
                    if(j-k <= 0) break;
                    dp[i][j] += dp[i-1][j-k];//第n枚骰子状态由第n-1枚转移而来
                }
            }
        }
        int count = pow(6, n);//所有点数出现的总次数
        vector<double> res;
        //n～6*n是可能产生的数，而且中间不会有空隙
        for(int i = n; i <= 6 * n; ++i) res.push_back(dp[n][i]*1.0/count);
        return res;
    }
};

可以看这位的题解配图：【n个骰子的点数】：详解动态规划及其优化方式 - n个骰子的点数 - 力扣（LeetCode）