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]
解法一:递归法,一个骰子有6个面,n个骰子的和一共有6n-n+1个值,类似于贪婪法,将所有的和都计算一遍,并将出现的和记录下来,组合一共有6^ n种,最后再将出现的计数除以6^n就是概率。
class Solution {
public:
vector<double> twoSum(int n) {
if(n <= 0)
return vector<double>(0,0.0);
vector<double> probability(6*n-n+1,0);
for(int i=1;i<=6;i++)
probabilityCore2(n,n,i,probability);
for(int i=0;i<probability.size();i++)
probability[i] = probability[i] / pow(6,n);
return probability;
}
void probabilityCore2(int originalNumber,int currentNumber,int sum,vector<double>& probability){
if(currentNumber == 1)
probability[sum-originalNumber]++; // 例如两个骰子的最小数2,在数组的第2-2个位置上
else{
for(int i=1;i<=6;i++)
probabilityCore2(originalNumber,currentNumber-1,sum+i,probability);
}
}
};
解法二:
使用动态规划 的方法(使用一维辅助数组):
p[n] = p[n-1]+p[n-2]+p[n-3]+p[n-4]+p[n-5]+p[n-6]
以两个骰子为例:
class Solution {
public:
vector<double> twoSum(int n) {
if(n <= 0)
return vector<double>(0,0.0);
vector<double> probability(6*n+1,0);
for(int i=1;i<=6;i++)
probability[i] = 1;
for(int i=2;i<=n;i++){ // i表示骰子的数量
for(int j=i*6;j>=i;j--){ // j表示要更新哪些和的计数
if(probability[j] != 0) // 将要计算的值归零
probability[j] = 0;
for(int k=1;k<=6;k++){
if(j-k > 0){
probability[j] += probability[j-k];
}
}
}
probability[i-1] = 0; // 注意将不可能得到的数归零
}
for(int j=n;j <= 6*n;j++)
probability[j] = probability[j] / pow(6,n);
return vector<double>(probability.begin()+n,probability.end());
}
};