剑指 Offer 60. n个骰子的点数
思路:动态规划
假设n-1个骰子的解为f(n-1),此时添加一枚骰子,n个骰子的点数和为x的概率f(n,x)为
f ( n , x ) = ∑ i = 1 6 f ( n − 1 , x − i ) × 1 6 f(n,x)=\sum\limits_{i=1}^6f(n-1,x-i)\times\frac{1}{6} f(n,x)=i=1∑6f(n−1,x−i)×61
如果按照此公式计算,会出现数组越界问题。
因此,考虑由f(n-1,x)推导f(n-1,x+i)
class Solution {
public:
vector<double> dicesProbability(int n) {
vector<double> dp(6,1.0/6.0);
for(int i=2;i<=n;++i){
vector<double> tmp(5*i+1,0);
for(int j=0;j<dp.size();++j){
for(int k=0;k<6;++k){
tmp[j+k]+=dp[j]/6.0;
}
}
dp=tmp;
}
return dp;
}
};
时间复杂度 O( n 2 n^2 n2) 计算的循环次数为 6 × 6 , 11 × 6 , [ 5 ( n − 1 ) + 1 ] × 6 6\times6,11\times6,[5(n-1)+1]\times6 6×6,11×6,[5(n−1)+1]×6总体时间复杂度为 O ( ( 6 + 5 ( n − 1 ) + 1 ) × ( n − 1 ) / 2 ) O((6+5(n-1)+1)\times(n-1)/2) O((6+5(n−1)+1)×(n−1)/2),即O( n 2 n^2 n2)
空间复杂度 O(n) 需要tmp辅助数组