下面代码是将次数换成概率计算,更方便。注意动态规划的思路!!
class Solution {
public:
vector dicesProbability(int n) {
vector ans;
double dp[15][70]={0};
//dp[i][j]: i个骰子,点数和为j的概率
//判断投1个骰子的情况,点数为i的概率是1/6
for(int i=1;i<=6;i++)
{
dp[1][i]=1.0/6.0;
}
//投2个以上的情况:[2,n]
for(int i=2;i<=n;i++)
{
// 抛第i个筛子所有点数的情况 [i, 6*i]
for(int j=i;j<=6*i;j++)
{
for(int k=1;k<=6;k++)
{
if(j-k<1)//防止出界
break;
//抛出第i个筛子点数为j的次数,即抛第i-1个筛子点数为j-k的次数
//(1 <= k <= 6),将这六种情况的次数都加起来
dp[i][j]+=dp[i-1][j-k]*1.0/6.0;
}
}
}
for(int p=n;p<=6*n;p++)
{
ans.push_back(dp[n][p]);
}
return ans;
}
};