这题出自LeetCode,题目如下:

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

你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

示例 1:

输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]

示例 2:

输入: 2
输出: [0.02778,0.05556,0.08333,0.11111,0.13889,0.16667,0.13889,0.11111,0.08333,0.05556,0.02778]

限制:

1 <= n <= 11

首先,容易发现这是一道动态规划的题,里面每一种可能性太多了,且为了计算每一种可能性,里面相互重叠的部分非常多,必然存在着大量重复的计算,而动态规划本质就是为了避免子可能性的重复计算而提出的算法。那么,接下来,我们要思考如何将问题进行拆分成子问题,并且通过解决子问题就可以快速得到问题本身的答案。

我们看到,虽然问题是去求概率,但是投掷n个骰子所能得到的不同排列次数是固定的,即

\[N = 6^n
\]

那么问题转换成用n个骰子组成和为s的不同排列次数有多少种,可以记为f(n,s)。此时,我们可以想象把这n个骰子分成两堆,一堆有n-1个,另一堆只有1个。显然,1个骰子掷出1的可能性只有1种,同样,掷出2、3、4、5、6的可能性也只有1种。那么问题转化成了:

\[f(n,s) = \sum_{i=1}^{6}f(n,s-i)
\]

容易知道,f(1,1),f(1,2),...f(1,6) = 1。我们可以先把这些边界值填上,然后不断地往上计算,就能得到最终的f(n,s)。代码如下:

class Solution {
public:
    vector<double> dicesProbability(int n) {
        int m = n * 6;
        int **a = new int *[m + 1];  
        for(int i = 0; i < m + 1; i++)  
        {  
            a[i] = new int[n + 1];
            for(int j = 0; j < n + 1; j++)  
            {
                a[i][j] = 0;
            }
        } 
        for(int i = 1; i <= m; i++)
        {
            if(i <= 6)
            {
                a[i][1] = 1;
            }
            else
            {
                a[i][1] = 0;
            }
        }

        for(int j = 2; j < n; j++)
        {
            a[1][j] = 0;
        }

        for(int i = 2; i <= m; i++)
        {
            for(int j = 2; j <= n; j++)
            {
                a[i][j] = calc(a, i, j);
            }
        }

        int all = 1;
        for(int j = 1; j <= n; j++)
        {
            all *= 6;
        }

        vector<double> res;
        for(int i = 1; i <= m; i++)
        {
            if(a[i][n] != 0)
            {
                res.push_back(a[i][n] / (all * 1.0));
            }
        }

        return res;
    }

    int calc(int **a, int m, int n)
    {
        int sum = 0;
        for(int i = 1; i <= 6; i++)
        {
            if(m - i > 0)
            {
                sum += a[m - i][n - 1];
            }
        }
        return sum;
    }
};

值得一提的是,可以看出我们的公式计算实际上只依赖前一个状态,所以其实并不需要二维存储空间,使用一维数组就能解决问题。

class Solution {
public:
    vector<double> dicesProbability(int n) {
        int *a = new int[6 * n + 1];
        for(int i = 0; i <= 6 * n; i++)
        {
            a[i] = 0;
        }
        for(int i = 1; i <= 6; i++)
        {
            a[i] = 1;
        }

        for(int i = 2; i <= n; i++)
        {
            for(int j = 6 * i; j >= i; j--)
            {
                int s = j - 6;
                if(s <= i - 1)
                {
                    s = i - 1;
                }
                a[j] = 0;
                for(int k = s; k <= j - 1; k++)
                {
                    a[j] += a[k];
                }
            }
        }

        vector<double> res;
        int all = 1;
        for(int i = 1; i <= n; i++)
        {
            all *= 6;
        }

        for(int i = n; i <= 6 * n; i++)
        {
            if(a[i] != 0)
            {
                res.push_back(a[i] / (all * 1.0));
            }
        }

        return res;
    }
};

注意一点,在计算f(n, s)时,数组前面6个f(n - 1, s - 1), ..., f(n - 1, s - 6)不一定都是有效的值,因为在计算f(n - 1, s)时,我们是从s = n - 1的情况下开始算的,这也好理解,毕竟n – 1个骰子掷出的最小点数之和必定是n – 1。所以我们在计算f(n, s)时要从第一个有效的值开始算起,即min(n - 1, s - 6)。否则可能会把前几次计算遗留下的脏值给错误计算进去,得到不正确的结果。

版权声明:本文为back-to-the-past原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://www.cnblogs.com/back-to-the-past/p/14159809.html