剑指offer-n个骰子的点数

LintCode

题目描述

扔 n 个骰子，向上面的数字之和为 S。给定 n，请列出所有可能的 S 值及其相应的概率。
思路
1.动态规划
空间复杂度是O(n^2)

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 
        //使用动态规划的方法
        //dp[n][m]存储的是n个骰子和为m的个数，它与n-1个骰子之间的关系是
        //dp[n][m] = dp[n-1][m-1]+dp[n-1][m-2]+...+dp[n-1][m-6]
        //dp[1][1]=dp[1][2]=...=dp[1][6]=1
        //n个骰子，它们的和的范围是n~6n
        long[][] dp = new long[n+1][6*n+1];//之前写的是int数组，然后测试有15个骰子的时候就报错了，因为超界了
        //动态数组初始化
        for(int i = 1;i <= 6;i++){
            dp[1][i] = 1;
        }
        for(int i = 2;i <= n;i++){
            for(int j = i;j <= 6*n;j++){
                for(int k = 1;k <= 6 && k<j;k++){
                    dp[i][j] += dp[i-1][j-k];
                }
            }
        }
        double total = Math.pow(6,n);
        List<Map.Entry<Integer,Double>> res = new ArrayList<>();
        for(int i = n;i <= 6*n;i++){
            res.add(new AbstractMap.SimpleEntry<>(i,dp[n][i]/total));
        }
        return res;
    }
}

2.将空间复杂度降低至O(n)

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 
        //使用动态规划的方法
        //dp[2][m]使空间复杂度降低至O(n)
      	//dp[0][m]来存储上一个骰子的次数，dp[1][m]是当前的次数
        long[][] dp = new long[2][6*n+1];//之前写的是int数组，然后15个骰子的时候就报错了，因为超界了
        //动态数组初始化
        for(int i = 1;i <= 6;i++){
            dp[0][i] = 1;
        }
        int m = 2;
        while(m <= n){
            for(int j = m;j <= 6*m;j++){
                for(int k = 1;k <= 6 && k<j;k++){
                    dp[1][j] += dp[0][j-k];
                }
            }
            for(int j = 0;j <= 6*m;j++){
                dp[0][j] = dp[1][j];
                dp[1][j]=0;
            }

            m++;
        }
        double total = Math.pow(6,n);
        List<Map.Entry<Integer,Double>> res = new ArrayList<>();
        for(int i = n;i <= 6*n;i++){
            res.add(new AbstractMap.SimpleEntry<>(i,dp[0][i]/total));
        }
        return res;
    }
}