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]

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

分析：

n个骰子所获得的点数之和存在于[n , 6*n]之间，并且可以知道n个骰子会有n^6种可能，因此很容易得到的一种解决方案为得到每个和出现的次数，再除以n^6即为结果。

假设process(n, sum) 返回n个骰子，结果为sum这种情况出现的个数。假设已经知道n-1的情况，该值存在以下6种可能，分别为第n个骰子点数为1, 2, 3, 4, 5, 6。因此把process(n - 1, sum - i)[i = 1 ->6]加起来即可。

递归方案：

    public double[] twoSum(int n) {
        //value n - > 6*n
        double[] result = new double[5*n + 1];
        int num = (int)Math.pow(6, n);
        for(int i = n; i < 6 * n + 1; i++){
            result[i - n] = (double)process(n, i) / num;
        }
        return result;
    }
    public int process(int n, int sum){
        if(sum < n || sum > 6 * n){
            return 0;
        }
        if(n == 1){
            return 1;
        }
        int result = 0;
        for(int i = 1; i <= 6; i++){
            result += process(n - 1, sum - i);
        }
        return result;
    }

动态规划:

发现上述递归方案具有极大的重复计算，由于process()结果只与n和sum的取值，并且无后效性。很容易想到将上述递归改成动态规划。

dp[i][j] 为 i个骰子投出j情况出现的次数。i =[1: n] ,j = [1: 6 * n]。

初值i =1 j=[1:6] dp[i][j] = 1;

 

需求dp[n][j] j = [n: 6 * n];

依次从上往下填表即可。

实现代码如下：

    public double[] twoSum(int n) {
        //value n - > 6*n
        double[] result = new double[5*n + 1];
        int num = (int)Math.pow(6, n);
        int[][] dp = new int[n + 1][6 *n + 1];
        //init
        for(int j = 1; j <= 6; j++){
            dp[1][j] = 1;
        }
        for(int i = 2; i <= n; i++){
            for(int j = i; j <= i * 6; j++){
                for(int d = 1; d <= 6; d++){
                    if(j - d < i - 1 || j - d > (i - 1) * 6){
                        continue;
                    }
                    dp[i][j] += dp[i - 1][j - d]; 
                }
            }
        }
        for(int i = n; i < 6 * n + 1; i++){
            result[i - n] = (double)dp[n][i] / num;
        }
        return result;
    }