【java】剑指offer60_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个骰子的解（即概率列表）为 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 - 1)可递推计算出 f(n)，而输入一个骰子的解 f(1) 已知，因此可通过解 f(1)依次递推出任意解 f(n)，递推为逆向会有边界越界问题，改为正向即可解决边界问题，即由于新增骰子的点数只可能为 1 至 6 ，因此概率 f(n - 1, x)仅与 f(n, x + 1) , f(n, x + 2), ... , f(n, x + 6)相关。因而，遍历 f(n - 1)中各点数和的概率，并将其相加至 f(n)中所有相关项，即可完成 f(n - 1)至 f(n) 的递推。

小tips：通常做法是声明一个二维数组 dp，dp[i][j]代表前 i个骰子的点数和 j的概率，并执行状态转移。而由于 dp[i] 仅由 dp[i-1]递推得出，为降低空间复杂度，只建立两个一维数组 dp , tmp交替前进即可。

public double[] dicesProbability(int n) {
        double[] dp = new double[6];
        Arrays.fill(dp, 1.0 / 6.0);
        for (int i = 2; i <= n; i++) {
            double[] tmp = new double[5 * i + 1];
            for (int j = 0; j < dp.length; j++) {
                for (int k = 0; k < 6; k++) {
                    tmp[j + k] += dp[j] / 6.0;
                }
            }
            dp = tmp;
        }
        return dp;
    }

复杂度分析：
时间复杂度 O(n ^ 2)： 状态转移循环 n - 1轮；每轮中，当 i = 2, 3, ..., n时，对应循环数量分别为6×6,11×6,...,[5(n−1)+1]×6 ；因此总体复杂度为 ，即等价于 O(n^2) 。
空间复杂度 O(n)： 状态转移过程中，辅助数组 tmp 最大长度为6(n−1)−[(n−1)−1]=5n−4 ，因此使用 O(5n - 4) = O(n)大小的额外空间。

作者：jyd
链接：https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof/solution/jian-zhi-offer-60-n-ge-tou-zi-de-dian-sh-z36d/