剑指 Offer 60. n个骰子的点数

1. 题目描述
n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
2. 思路
（1）设dp[i][j]为第i枚骰子，点数j出现的次数,则最后需要求的结果为所有点数各自出现的次数除以总的次数（6^n）；
（2）转移方程为原来的点数加上第i次出现的点数之和。
（3）初始化次数为1的情况。
（4）求出概率。

3. 代码

class Solution {
	public double[] dicesProbability(int n) {
        //设dp[i][j]为第i枚骰子，点数j出现的次数
        //则最后需要求的结果为所有点数出现的次数各自除以总的次数（6^n）
        int[][] dp = new int[n+1][6*n+1];
        //初始化次数为1的情况
        for(int j = 1; j<=6; j++){
            dp[1][j] = 1;
        }

        for(int i = 2; i<=n; i++){
            //注意从i开始，因为最小点数为i
            for(int j = i; j<=6*i; j++){
                for(int k = 1; k<=6; k++){
                    if(j-k<=0)
                        break;
                        //原来的dp[i][j]=0
                    dp[i][j] += dp[i-1][j-k];//转移方程
                }
            }
        }
        
        double all = Math.pow(6,n);
        //总的点数为：最大点数-最小点数+1
        double[] res = new double[6*n-n+1];
        int j=0;
        //求概率
        for(int i=n; i<=6*n; i++){
            res[j++] = dp[n][i]/all;
        }
        return res;
    }
}