剑指 Offer 60. n个骰子的点数
难度中等197收藏分享切换为英文接收动态反馈
把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
题解
这里采用动态规划,思想有点复杂,就是采用正向递推,思想如最下方图
class Solution {
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++) {
//这里是一个骰子的6种情况
for (int k = 0; k < 6; k++) {
tmp[j + k] += dp[j] / 6.0;
}
}
//这里进行扩容,进一步计算
dp = tmp;
}
return dp;
}
}
