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]

class Solution {
        public double[] dicesProbability(int n) {
        double[][] dp = new double[n + 1][n * 6 + 1];
        //i横坐标代表带塞子的个数
        //j代表当前需要凑的点数
        //如果有2个塞子,那么第二个塞子,如果点数是2的话,最后一个塞子是1的话,那么我需要上一个塞子点数是1。如果点数和是4的话,那么最后一个塞子如果是1,2,3:那么上一个塞子一定是3,2,1的概率和,但是如果上一个塞子是3,2,1,那么最后一个塞子一定是1,2,3,。那么就需要一个塞子是1、2、3的概率和。因为确定的前面的塞子,最后一颗塞子一定是确定的。画图,即可
            /*
            塞子的点数和  1 2 3 4 5 6  7  8  9  10 11 12 13 14 15
            塞子的个数
            1            1 1 1 1 1 1
            2              1 2 3 4 5  6  5  4  3  2  1
            3                1 3 6 10 15 21 25 27
            每个二维数组的值/Math.pow(6,塞子的个数)
            */
        //dp[i][j]=代表i个塞子,摇到j的概率
        //第一个,一定是1
        //从第二个开始,每次加上当前行前面的,并且加上上一行,上一列的
        //如果个数够6个了,那么就要减去deque.removeFirst()
        //最后一个塞子,只可能是1-6,6种可能,所以求出前面点数和,塞子个数-1的概率即可
        Deque<Double> deque = new LinkedList<>();//从尾巴进,从头出,最多只留6个
        for (int i = 1; i <= 6; i++) {
            dp[1][i] = 1.0;
        }
        for (int i = 2; i <= n; i++) {
            deque.clear();
            int left = i;
            int right = i * 6;
            dp[i][left++] = 1.0;
            dp[i][right--] = 1.0;
            deque.addLast(1.0);
            while (left <= right) {
                if (deque.size() != 6) {
                    deque.addLast(dp[i - 1][left - 1]);
                    dp[i][left] = dp[i][left - 1] + dp[i - 1][left - 1];
                } else {
                    deque.addLast(dp[i - 1][left - 1]);
                    dp[i][left] = dp[i][left - 1] + dp[i - 1][left - 1] - deque.removeFirst();
                }
                dp[i][right] = dp[i][left];
                left++;
                right--;
            }
        }
        double[] res = new double[n * 6 - n + 1];
        int index = 0;
        for (int i = n; i < n * 6 + 1; i++) {
            res[index++] = dp[n][i] / Math.pow(6, n);
        }
        return res;
    }
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值