题目描述
扔 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;
}
}