[LintCode]Dices Sum

原创于 2017-04-17 20:56:15 发布 · 827 阅读

0 ·

CC 4.0 BY-SA版权

LintCode 专栏收录该内容

24 篇文章

订阅专栏

本文介绍了一种使用动态规划算法解决骰子投掷n次后各种和出现概率的问题。通过构建一个二维数组dp来存储每次投掷得到特定总和的概率，并逐步计算出最终的概率分布。

http://www.lintcode.com/en/problem/dices-sum/

骰子投掷n次，求所有情况的和出现的概率

二维数组dp[i][j]保存投掷i次得到和为j的概率，当前位置的概率为当前投掷1 ~ 6的情况下的前序概率和

public class Solution {
    /**
     * @param n an integer
     * @return a list of Map.Entry<sum, probability>
     */
    public List<Map.Entry<Integer, Double>> dicesSum(int n) {
        // Write your code here
        // Ps. new AbstractMap.SimpleEntry<Integer, Double>(sum, pro)
        // to create the pair
        List<Map.Entry<Integer, Double>> res = new LinkedList();
        double[][] dp = new double[n + 1][6 * n + 1];
        for (int i = 1; i <= 6; i++) {
            dp[1][i] = 1.0 / 6.0;
        }
        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) {
                        dp[i][j] += dp[i - 1][j - k];
                    }
                }
                dp[i][j] /= 6.0;
            }
        }
        for (int i = n; i <= 6 * n; i++) {
            res.add(new AbstractMap.SimpleEntry<Integer, Double>(i, dp[n][i]));
        }
        return res;
    }
}