n个骰子的点数

题目

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

分析

由于每个dp[i][j]只于i-1时刻的状态有关，所以可以删去一个维度，简化算法,具体优化见代码。
dp[i]代表点数为i时的排列情况总数。
dp[12]=dp[11]+dp[10]+dp[9]+dp[8]+dp[7]+dp[6],

j表示的点数，k表示的是一个骰子的数字，j-k表示的是去掉一个骰子的点数，i表示的是骰子个数，i-1表示的是去掉一个骰子的个数，一个骰子最小的数字是1，j-k<i-1表示的是（j-k）去掉一个骰子的点数不可能小于（i-1）少一个骰子情况下的最小值，类似不可能存在dp[2][1]这种东西。

代码

class Solution {
    public double[] dicesProbability(int n) {
        int[][] dp = new int[n+1][6*n+1];
        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 k=1;k<=6;k++){
                    if(j<=k) break;
                    dp[i][j] +=dp[i-1][j-k];
                }
            }
        }
        double[] res = new double[5*n+1];
        int index =0;
        double sum = Math.pow(6, n);
        for(int i=n;i<=6*n;i++){
            res[index++] = dp[n][i]/sum;
        }
        return res;
    }
}

优化后：

class Solution {
    public double[] dicesProbability(int n) {
       // 总的状态数为6 * n，bp[0]不使用
        int[] dp = new int[6 * n + 1];
        // 总的次数
        double all = Math.pow(6, n);

        //阶段1的状态
        for(int i = 1; i <= 6; i++) {
            dp[i] = 1;
        }

        for(int i = 2; i <= n; i++) {
            // 阶段i的状态，与阶段i-1的状态有关系，所以j从i*6开始
            for(int j = i * 6; j >= i; j--) {
                dp[j] = 0;
                for(int k = 1; k <= 6; k++) {
                    if(j - k < i - 1) break;
                    dp[j] += dp[j - k];
                }
            }
        }

        double[] ans = new double[6 * n - n + 1];
        // 计算概率
        for(int i = n; i <= n * 6; i++) {
            ans[i - n] = dp[i] / all;
        }

        return ans;

    }
}