面试题60:n个骰子出现的概率
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
class Solution {
public:
vector<double> dicesProbability(int n) {
vector<double> res(n*5+1); // n个骰子共出现n~6n,共5n+1种情况;
vector<vector<double> > dp(n+1,vector<double>(n*6+1)); // 创建一个(n+1)x(6n+1)二维数组表示概率,数组占大空间方便下来调用明确
for(int i=1;i<=6;i++) // 只有一个骰子的情况
{
dp[1][i] = 1.0/6.0;
}
for(int i=2;i<=n;i++) // n个骰子的情况,i从第2个骰子开始
{
for(int j=i;j<=i*6;j++) // j共出现i~6i中情况
{
for(int k=1;k<=6;k++) // 最后一个骰子k分别1~6的情况
{
if(j-k>0) // eg:共2个骰子,i=2,j=2,k从1开始;计算出现2的概率:(1,1)的情况有效,(2,0)、(3,-1)...的情况无效。目的是确保数组不过界。
dp[i][j] += dp[i-1][j-k] / 6.0; // 当下概率值只和另一个(另几个)骰子的概率有关;
else
break;
}
}
}
for(int i=0;i<n*5+1;i++) // 给res数组赋值;
res[i] = dp[n][n+i];
return res;
}
};