剑指Offer 面试题60. n个骰子的点数（Java代码）

https://leetcode-cn.com/problems/nge-tou-zi-de-dian-shu-lcof/

题目描述

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

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

输入输出样例

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

题解

动态规划：

N个骰子的结果，由N-1个骰子的结果 与 1 个骰子的结果相绑定

• 对于N-1个骰子，其结果中的每一个概率和 1个骰子的结果进行累加，加过添加到另一个数组中，这样得到的就是 N个骰子的结果

class Solution {
    public double[] twoSum(int n) {
        double[] pre = {1/6d, 1/6d, 1/6d, 1/6d, 1/6d, 1/6d};

        // 自底而上地考虑问题
        for(int i=2; i<=n; i++){
            double[] tmp = new double[i*5 + 1];
            for(int p=0; p<pre.length; ++p){
                for(int j=0; j<6; j++){
                    tmp[p+j] += pre[p]/6;
                }
            }
            pre = tmp;
        }

        return pre;
    }
}