剑指 Offer 60题：n个骰子的点数

1.题目描述

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

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

示例1：

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof

2.思路分析

定义dp数组： dp[i][j]表示把i个骰子扔地下，得到和为j的概率

初始化：当只有一个骰子时，和为1,,,6的概率都是 1 / 6

for(int i = 1; i <= 6; i++){

       dp[1][ i ] = 1.0 / 6;

}

状态转移方程：

假设已知 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) 。递推公式如下所示：

3.解答 

class Solution {
    //DP

    public double[] dicesProbability(int n) {
        //定义dp数组：dp[i][j]表示把i个骰子扔地下，得到和为j的概率
        double[][] dp = new double[n + 1][6 * n + 1];
        //初始化
        for(int i = 1; i <= 6; i++){
            dp[1][i] = 1.0 / 6;
        }
        //状态转移
        for(int i = 2; i <= n; i++){
            for(int j = i; j <= 6 * i; j++){
                //增加的一个骰子可能出现的点数是1，，，6
                for(int k = 1; k <= 6; k++){
                    if(j - k > 0){
                        //判断是否越界
                        dp[i][j] += dp[i - 1][j - k] / 6; 
                    }else{
                        break;
                    }
                }
            }
        }

        //计算结果
        double[] res = new double[5 * n + 1];
        for(int i = 0; i <= 5 * n; i++){
            res[i] = dp[n][n + i];
        }

        return res;
    }
}

时间复杂度：O(N * N)

空间复杂度：O(N * N)