算法---LeetCode 剑指 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]

示例 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
👍 238 👎 0

2. 题解

2.1 解法1: 动态规划

1. 定义状态数组: 分析问题的状态时，不要分析整体，只分析最后一个阶段即可
   dp[i][j] ，表示投掷完 i 枚骰子后，点数 j 的出现次数。

2. 状态转移方程: 分析最后一个阶段，分析它的状态是如何得到的。
   dp[n][j]来表示最后一个阶段点数 j 出现的次数。
   单单看第 n 枚骰子，它的点数可能为 1 , 2, 3, … , 6，因此投掷完 n 枚骰子后点数 j 出现的次数，可以由投掷完 n-1 枚骰子后，对应点数 j-1, j-2, j-3, … , j-6 出现的次数之和转化过来。

for (第n枚骰子的点数 i = 1; i <= 6; i ++) {
	dp[n][j] += dp[n-1][j - i]
}

n 表示阶段，j 表示投掷完 n 枚骰子后的点数和，i 表示第 n 枚骰子会出现的六个点数。

3. 边界处理:
   第一阶段的状态：投掷完 11 枚骰子后，它的可能点数分别为 1, 2, 3, … , 6，并且每个点数出现的次数都是 1

for (int i = 1; i <= 6; i ++) {
dp[1][i] = 1;
}

    class Solution {
        public double[] dicesProbability(int n) {
            int[][] dp = new int[12][67];
            // 初始化一个骰子的情况
            for (int i = 1; i <= 6; i++) {
                dp[1][i] = 1;
            }

            // 第一层循环: 骰子个数
            for (int i = 2; i <= n; i++) {
                // 第二层循环, 投出的总点数
                for (int j = i; j <= 6 * i; j++) {
                    // 第三重循环, 为了找到能得到这种可能点数时，对于少一个色子时的所有可能情况，相加即是当前可能总和
                    for (int curr = 1; curr <= 6; curr++) {
                        // i-1,即是对于只有对于少一个色子时，最小和最少为i-1，不可能再小了
                        if (j - curr < i - 1) {
                            break;
                        }
                        // 计算, 上一个状态的点数累加
                        dp[i][j] += dp[i - 1][j - curr];
                    }
                }
            }
            // 计算各种点数出现的总次数
            double all = Math.pow(6, n);
            // 数组长度为所有可能出现的次数种类
            double[] ans = new double[6 * n - n + 1];

            for (int i = n; i <= n * 6; i++) {
                ans[i - n] = (double) dp[n][i] / all;
            }
            return ans;
        }
    }

参考: 【n个骰子的点数】：详解动态规划及其优化方式