[剑指 offer] --动态规划--面试题60. n个骰子的点数

1 题目描述

把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]

限制：

1 <= n <= 11

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

2 解题思路

• 方法：动态规划
  n个骰子，一共有6**n种情况
  n=1, 和为s的情况有 F(n,s)=1 s=1,2,3,4,5,6
  n>1 , F(n,s) = F(n-1,s-1)+F(n-1,s-2) +F(n-1,s-3)+F(n-1,s-4)+F(n-1,s-5)+F(n-1,s-6)
  可以看作是从前(n-1)个骰子投完之后的状态转移过来，
  考虑用动态规划
  ————————————————
  版权声明：本文为CSDN博主「0607offer」的原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接及本声明。
  原文链接：https://blog.csdn.net/qq_24243877/article/details/104560944

3 解决代码

• 方法：动态规划《Java代码》

class Solution {
    public double[] twoSum(int n) {
        表示i个骰子掷出s点的次数
        int dp[][] = new int[n + 1][6*n + 1];
        for(int i = 1; i <= 6; i++){
            表示一个骰子掷出i点的次数为1
            dp[1][i] = 1;
        }
        for(int i = 2; i <= n; i++){//表示骰子的个数
            for(int s = i; s <= 6*i; s++){//表示可能会出现的点数之和
                for(int j = 1; j <= 6; j++){//表示当前这个骰子可能掷出的点数
                    if(s - j < i- 1){//当总数还没有j大时，就不存在这种情况
                        break;
                    }
                    //当前n个骰子出现的点数之和等于前一次出现的点数之和加上这一次出现的点数
                    dp[i][s] += dp[i - 1][s - j];
                }
            }
        }
        //掷出n次点数出现的所有情况
        double total = Math.pow((double)6, (double)n);
        //因为最大就只会出现5*n+1中点数
        double[] ans = new double[5 * n + 1];
        for(int i = n; i <= 6*n; i++){
            //第i小的点数出现的概率
            ans[i - n] = (double)dp[n][i] / total;
        }
        return ans;

    }
}