Interview100-67.2 N个骰子和为S的问题

最新推荐文章于 2022-10-30 08:00:00 发布

Ethan_pika

最新推荐文章于 2022-10-30 08:00:00 发布

阅读量197

点赞数

分类专栏：面试算法100题文章标签： algorithm interview

本文链接：https://blog.csdn.net/yitian_z/article/details/88175543

版权

面试算法100题专栏收录该内容

50 篇文章 0 订阅

订阅专栏

题目

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

解法

1. 首先这个问题本身要求的时n个骰子朝上的点数之和为S的这种取值的概率P。P=和为S出现的次数 / 所有点数之和的可能性。由于骰子的点数只有1到6，所以n个骰子的所有点数之和的可能性为6的n次方。这样这个问题就可以转换为求n个骰子点数之和为S的所有可能取值的问题，然后再除以6的n次方，即为可能值出现的概率。

2. 对于n个骰子点数向上和为s的问题，使用F(s,n)表示n个骰子和为s的所有可能取值情况。例如当n=1时，只有一个骰子，那么f(s, 1) = f(1, 1) + f(2, 1) + f(3, 1) + f(4, 1) + f(5, 1) + f(6, 1) = 1 x 6 = 6。即一个骰子的情况下，s的可能取值为1，2，3，4，5，6。所以，当n个骰子时，第n个骰子向上的点数有可能的取值为1，2，3，4，5，6，所以有如下情况：

当第n个骰子向上的点数取值为1时，f(s, n) = f(s-1, n-1)
当第n个骰子向上的点数取值为2时，f(s, n) = f(s-2, n-1)
当第n个骰子向上的点数取值为3时，f(s, n) = f(s-3, n-1)
当第n个骰子向上的点数取值为4时，f(s, n) = f(s-4, n-1)
当第n个骰子向上的点数取值为5时，f(s, n) = f(s-5, n-1)
当第n个骰子向上的点数取值为6时，f(s, n) = f(s-6, n-1)

所以，f(s, n) = f(s-1, n-1) + f(s-2, n-1) + f(s-3, n-1) + f(s-4, n-1) + f(s-5, n-1) + f(s-6, n-1)，而且 f(1, 1) = f(2, 1) = f(3, 1) = f(4, 1) = f(5, 1) = f(6, 1) = 1。因此这个问题可以使用递归和动态规划进行实现。

递归方式

N-S-PROBLEM(int n, int s):
    if n < 1 or s > 6n or s < n:
        return;
    if n == 1:
        return 1;
    int result_num = 0;
    result_num = N-S-PROBLEM(n-1, s-1) + N-S-PROBLEM(n-1, s-2) + N-S-PROBLEM(n-1, s-3) + N-S-PROBLEM(n-1, s-4) + N-S-PROBLEM(n-1, s-5) + N-S-PROBLEM(n-1, s-6);
    return result_num;

动态规划

这里放一个代码，引自：https://blog.csdn.net/GarfieldEr007/article/details/50722219

用两个数组来存储骰子点数的每一种出现的次数。
在一次循环中，第一个数组中的第n个数字表示骰子和为n出现的次数。
在下一次循环中我们加上一个新的骰子，此时和为n的骰子出现的次数应该等于上一次循环中骰子点数和为n-1、n-2、n-3、n-4、n-5与n-6的次数的综合，所以我们把另一个数组的第n个数字设为前一个数组对应的第n-1、n-2、n-3、n-4、n-5与n-6之和。

void PrintSumProbabilityOfDices_2(int number)
{
    double* pProbabilities[2];
    pProbabilities[0] = new double[g_maxValue * number + 1];
    pProbabilities[1] = new double[g_maxValue * number + 1];
    for(int i = 0; i < g_maxValue * number + 1; ++i)
    {
        pProbabilities[0][i] = 0;
        pProbabilities[1][i] = 0;
    }
 
    int flag = 0;
    for (int i = 1; i <= g_maxValue; ++i)
        pProbabilities[flag][i] = 1;
      
    for (int k = 2; k <= number; ++k)
    {
        for (int i = k; i <= g_maxValue * k; ++i)
        {
            pProbabilities[1 - flag][i] = 0;
            for(int j = 1; j <= i && j <= g_maxValue; ++j)
                pProbabilities[1 - flag][i] += pProbabilities[flag][i - j];
        }
 
        flag = 1 - flag;
    }
 
    double total = pow((double)g_maxValue, number);
    for(int i = number; i <= g_maxValue * number; ++i)
    {
        double ratio = pProbabilities[flag][i] / total;
        printf("%d: %f\n", i, ratio);
    }
 
    delete[] pProbabilities[0];
    delete[] pProbabilities[1];
}