【剑指Offer 60. n个骰子的点数】动态规划+概率

【剑指Offer 60. n个骰子的点数】动态规划+概率_第1张图片

【剑指Offer 60. n个骰子的点数】动态规划+概率_第2张图片

下面代码是将次数换成概率计算,更方便。注意动态规划的思路!!

class Solution {
public:
    vector dicesProbability(int n) {
        vector ans;
        double dp[15][70]={0};
        //dp[i][j]: i个骰子,点数和为j的概率
        //判断投1个骰子的情况,点数为i的概率是1/6
        for(int i=1;i<=6;i++)
        {
            dp[1][i]=1.0/6.0;
        }
        //投2个以上的情况:[2,n]
        for(int i=2;i<=n;i++)
        {
            // 抛第i个筛子所有点数的情况 [i, 6*i]
            for(int j=i;j<=6*i;j++)
            {
                for(int k=1;k<=6;k++)
                {
                    if(j-k<1)//防止出界
                        break;
                    //抛出第i个筛子点数为j的次数,即抛第i-1个筛子点数为j-k的次数 
                    //(1 <= k <= 6),将这六种情况的次数都加起来
                    dp[i][j]+=dp[i-1][j-k]*1.0/6.0;
                }
            }
        }
        for(int p=n;p<=6*n;p++)
        {
            ans.push_back(dp[n][p]);
        }
        return ans;
    }
};

你可能感兴趣的:(保研刷题,动态规划,算法,leetcode,c++)