(剑指Offer)面试题43:n个骰子的点数

题目:

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

思路:

s可能出现的值的范围为:n–6*n

1、全排列

回溯法枚举n个骰子(6面)的全排列,然后计算每一次排列所有值的和,并统计该和的出现的次数,除以6^n(全排列的全部可能性),即为概率。(这里就不列出代码)

2、递归思想

通过递归的思想将n个骰子的点数累加。

要求出n个骰子的点数和,可以先求出前n-1个骰子的点数和,然后加上第n个骰子的点数;

递归结束条件:n=1,此时某个点数和出现的次数+1;

3、动态规划思想

假设f(m,n)表示投第m个骰子时,点数之和n出现的次数,投第m个骰子时的点数之和只与投第m-1个骰子时有关。

递归方程:f(m,n)=f(m-1,n-1)+f(m-1,n-2)+f(m-1,n-3)+f(m-1,n-4)+f(m-1,n-5)+f(m-1,n-6),表示本轮点数和为n出现次数等于上一轮点数和为n-1,n-2,n-3,n-4,n-5,n-6出现的次数之和。

初始条件:第一轮的f(1),f(2),f(3),f(4),f(5),f(6)均等于1.

代码:

1、递归方法

#include <iostream>
#include <math.h>

using namespace std;

int g_maxValue=6;

void Probability(int original,int index,int curSum,int* pProbability){
    if(index==0){
        pProbability[curSum-original]+=1;
        return;
    }
    for(int i=1;i<=6;i++)
        Probability(original,index-1,curSum+i,pProbability);
}

void PrintProbability(int n){
    if(n<1)
        return;
    int maxSum=n*g_maxValue;
    int* pProbability=new int[maxSum-n+1];
    for(int i=n;i<=maxSum;i++)
        pProbability[i-n]=0;
    int curSum=0;
    Probability(n,n,curSum,pProbability);

    int total=pow((double)g_maxValue,n);
    double prob=0;
    for(int i=n;i<=maxSum;i++){
        double ratio=(double)pProbability[i-n]/total;
        prob+=ratio;
        cout<<i<<" "<<ratio<<" "<<endl;
    }
    cout<<prob<<endl;
    cout<<endl;

    delete[] pProbability;
}

int main()
{
    int n=5;
    PrintProbability(n);
    return 0;
}

2、动态规划

#include <iostream>
#include <math.h>

using namespace std;

int g_maxValue=6;
void PrintProbability(int n){
    if(n<1)
        return;

    int* pProbability[2];
    pProbability[0]=new int[g_maxValue*n+1];
    pProbability[1]=new int[g_maxValue*n+1];

    for(int i=0;i<=g_maxValue*n;i++){
        pProbability[0][i]=0;
        pProbability[1][i]=0;
    }

    int flag=0;
    for(int i=1;i<=g_maxValue;i++)
        pProbability[flag][i]=1;

    for(int k=2;k<=n;k++){
        for(int i=0;i<k;i++)
            pProbability[1-flag][i]=0;
        for(int i=k;i<=g_maxValue*k;i++){
            pProbability[1-flag][i]=0;
            for(int j=1;j<=i && j<=g_maxValue;j++)
                pProbability[1-flag][i]+=pProbability[flag][i-j];
        }
        flag=1-flag;
    }

    int total=pow((double)g_maxValue,n);
    double prob=0;
    for(int i=0;i<=g_maxValue*n;i++){
        //cout<<pProbability[flag][i]<<endl;
        double ratio=(double)pProbability[flag][i]/total;
        prob+=ratio;
        cout<<i<<" "<<ratio<<" "<<endl;
    }
    cout<<prob<<endl;
    cout<<endl;

    delete[] pProbability[0];
    delete[] pProbability[1];
}

int main()
{
    int n=5;
    PrintProbability(n);
    return 0;
}

  

    原文作者:AndyJee
    原文地址: https://www.cnblogs.com/AndyJee/p/4686208.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞