【LeetCode】【动态规划】剑指 Offer 60. n个骰子的点数 思路解析和代码

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

题目链接

个人思路

题意

要理解n个骰子是同时掷出，即最小值的点数是n，最大的点数是6 * n，共有6ⁿ种可能
尤其注意最小值是n而不是1，不是将骰子一个接一个掷出去的

思路

• 重叠子问题：n个骰子，分为当前一个骰子（1 ~ 6）与其余n-1骰子的点数之和相加；n-1个骰子，可以分为当前一个骰子（1~6）与其余n-2个骰子的点数之和相加
  dp[i][j]表示：投掷i个骰子，得到点数j的次数是dp[i][j]
• 边界：第一颗骰子只有0~6，dp[1][i] = 1 i的取值为1~6
• 递推方程：

for(int i = 2; i <= n; ++i){//n个骰子
    for(int j = i; j <= 6 * i; ++j){//n个骰子能组合出来的点数
        for(int k = 1; k <= 6; ++k){//当前骰子的点数
            if(j - k <= 0){
                continue;
            }
            dp[i][j] += dp[i - 1][j - k];
        }
    }
}

个人思路代码

class Solution {
public:
    int dp[15][105];//dp[i][j]表示投掷i个骰子，得到点数j的次数是dp[i][j]
    vector<double> dicesProbability(int n) {
        vector<double> ans;
        //边界
        for(int i = 1; i <= 6; ++i){
            dp[1][i] = 1;
        }

        for(int i = 2; i <= n; ++i){//n个骰子
            for(int j = i; j <= 6 * i; ++j){//n个骰子能组合出来的点数
                for(int k = 1; k <= 6; ++k){//当前骰子的点数
                    if(j - k <= 0){
                        continue;
                    }
                    dp[i][j] += dp[i - 1][j - k];
                }
            }
        }

        int total = pow(6, n);
        for(int i = n; i <= 6 * n; ++i){
            ans.push_back(dp[n][i] * 1.0 / total);
        }
        return ans;
    }
};