剑指Offer：[第29天 动态规划(困难)]---＞n个骰子的点数

---

一、题目描述

把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$ 个骰子，可得：
----每个骰子摇到 $1$ 至 $6$ 的概率相等，都为 $\frac{1}{6}$
----将每个骰子的点数看作独立情况，共有 $6^n$ 种「点数组合」。例如 $n=2$ 时的点数组合为：
$(1,1),(1,2),(1,3)\cdots ,(2,1),(2,2),(2,3)\cdots ,(6,1),\cdots ,(6,6)$
----$n$ 个骰子「点数和」的范围为 $[n,6n]$，数量为 $6n-n+1=5n+1$种。
输入 $n=2$ 时，点数组合、点数和、各点数概率的计算过程如下：

动态规划：
----设输入 $n$ 个骰子的解(即概率列表)为 $f(n)$，其中「点数和」$x$ 的概率为 $f(n,x)$。
----假设已知 $n-1$ 个骰子的解 $f(n-1)$，此时添加一枚骰子，求 $n$ 个骰子的点数和为 $x$ 的概率 $f(n,x)$。当添加骰子的点数为 $1$ 时，前 $n-1$ 个骰子的点数和应为 $x-1$，方可组成点数和 $x$。同理，当此骰子为 $2$ 时，前 $n-1$ 个骰子应为 $x-2$。以此类推，直至此骰子点数为 $6$。将这 6 种情况的概率相加，即可得到概率 $f(n,x)$。递推公式为：$f(n,x)=\sum _{i=1}^{6}f(n-1,x-i)\times \frac{1}{6}$
----根据以上分析，得知通过子问题的解 $f(n-1)$ 可递推计算出 $f(n)$，而输入一个骰子的解 $f(1)$ 已知，因此可通过解 $f(1)$ 依次递推出任意解 $f(n)$
----如下图所示，为 $n=2,x=7$ 的递推计算示例。

算法流程：
①初始化：建立一个 $(n+1)\times (6n+1)$的状态转移二维数组 $dp$、建立一个大小为 $5n+1$ 用于返回结果的数组 $res$
②循环处理：第1层循环用于遍历骰子数量，第2层循环用于遍历当前骰子的点数和，第3层循环用于遍历当前骰子点数和如何由少一个骰子转移而来
③返回：将 $dp[n]$ 上的各个元素赋值给 $res$，并返回 $res$。
复杂度分析：
时间复杂度$\mathrm{O}({\mathrm{n}}^2)$：第一层循环 $n-1$ 次。第二层循环中，当 $i=2,3,...,n$ 时，对应循环数量分别为 $11,16,...,5n+1$。第三层循环6次。因此总体复杂度为 $O(n^2)$。
空间复杂度$\mathrm{O}({\mathrm{n}}^2)$：这里二维数组 dp 的占用 $O(n^2)$ 大小空间。
注意： 这里空间复杂度还可以优化，由于 $dp[i]$ 仅由 $dp[i-1]$ 得出，可以只建立两个一维数组 $dp$、$tmp$ 交替前进即可。

三、整体代码

整体代码如下

/**
 * Note: The returned array must be malloced, assume caller calls free().
 */

double* dicesProbability(int n, int* returnSize){
    *returnSize = n * 5 + 1;   //可能出现的结果数量
    double dp[n+1][n*6+1];     //状态转移数组
    double *res = (double*)malloc(sizeof(double) * (n*5+1));  //返回结果的数组
    memset(dp, 0, sizeof(dp));                 
    dp[0][0] = 1;

    //骰子数为1时候，每面出现的概率都是1/6
    for(int i = 1; i <= 6; i++){
        dp[1][i] = 1.0/6.0;
    }

    for(int i = 2; i <= n; i++){          //第1次循环，骰子的数量
        for(int j = i; j <= i*6; j++){    //第2次循环，当前骰子的点数和
            for(int k = 1; k <= 6; k++){  //第3次循环，当前骰子的点数和由i-1个骰子如何转移而来
                if(j - k > 0)     //确保不越界
                    dp[i][j] += dp[i-1][j-k]/6.0; 
                else
                    break;  
            }

        }
    }
    
    //进行赋值
    for(int i = 0; i <= 5 * n; i++){
        res[i] = dp[n][n+i];
    }

    return res;
}

运行，测试通过