剑指offer--c++--n个骰子的点数

题目:

          把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。

感谢大佬的帮助:https://www.cnblogs.com/wangkundentisy/p/9378886.html

题目分析:

       这是一题典型的DP问题。

    1. 首先分析是否是DP问题,得满足两个条件:1.最优子结构;2.重叠子问题。

       第n个骰子的点数实际上是在n-1这堆骰子的基础上进行的投掷。满足最优子结构

       求父问题的时候多次用到子问题,想要解决第n个骰子的问题,需要前面的数据。满足重叠子问题

    2.确定是DP问题之后,需要确定状态转移方程。

      分析一下,第n个骰子的点数之和是与前面n-1那堆骰子的点数之和相关。其实情况还算蛮复杂的,光想是想不出来的,我们根据色子的投掷情况,列了下面表格,根据观察我们发现状态转换方程。我们不妨设f(n,k) = f(n-1,k-1) + f(n-1,k-2) + f(n-1,k-3) + f(n-1,k-4) + f(n-1,k-5) + f(n-1,k-6) ;这里n代表色子个数,k代表点数之和,f(n,k)代表n个色子投掷,点数之和为k的所有投掷次数。仔细想一下,第n个骰子投掷完之后的点数之和是不是等于上一次投掷点数次数之和,因为上一次有六种投法,所以要将这六种的可能次数都加起来。

   3.在建立完状态方程之后,考虑将状态方程初始化,第一次投掷,1~6点数各自出现的次数都为1次吧,所以都初始化为1。

最后编写代码:

class Solution {
public:
    vector<double> twoSum(int n) {
       vector<double> res;
       
       vector<vector<int> > dp(n+1,vector<int>(n*6+1,0));

       //初始化
       for(int i = 1; i <= 6; i++)
       {
           dp[1][i] = 1;
       }

       //填表
       int sum = 0;
        for(int i = 2; i <= n; i++)
        {
            for(int j = i; j <= n*6; j++)
            {
                for(int k = 1; k <= 6; k++)
                {
                    if((j-k) >= 0)//这里防止数组越界,看表就知道,数组会越界的
                        dp[i][j] += dp[i-1][j-k];
                    else
                        break;
                }
                if(i == n)
                {
                    sum += dp[i][j];//直接计算n个色子投掷所有可能出现的次数总和
                }
            }
        }
        if(n == 1)
            sum = 6;
        
        for(int i = n; i <= n*6; i++)
        {
            res.push_back((double)dp[n][i]/sum);//用于计算概率,别忘记强制转换成double
        }

        return res;
    }
};

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值