剑指offer n个骰子点数出现的概率

把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

image.png

此题可采用动态规划的方法解决。

n个骰子共会出现6n-n+1,即5n+1种情况。
编写代码如下

 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[] temp = new  double[5*i+1];
            for(int j = 0;j<dp.length;j++){
                for(int k = 0 ; k<6 ; k++){
                    temp[j+k] += dp[j]*1.0/6.0;
                }
            }
            dp = temp;
        }
        return dp;
    }
}
 

Q.E.D.