剑指 Offer 60. n个骰子的点数 - Touale Cula's Blog

题目内容

把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。

 

你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

 

示例 1:

1
2
输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]

示例 2:

1
2
输入: 2
输出: [0.02778,0.05556,0.08333,0.11111,0.13889,0.16667,0.13889,0.11111,0.08333,0.05556,0.02778]

限制:

1
1 <= n <= 11

解法一:动态规划

// f(n,s) = sum(f(n-1,s-1),…f(n-1,s-5)) * 1/6

第n个骰子和第n-1个骰子状态有关

比如只有两个骰子的时候,要求投出7的概率等于
1,6
2,5
3,4
4,3
5,2
6,1

即:f(2,7) = (f(1,1)+f(1,2)+f(1,3)+f(1,4)+f(1,5)+f(1,6))*(1/6)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
class Solution {
public:
vector<double> dicesProbability(int n) {

vector<double> dp(6,1.0/6.0);
for(int i = 1;i<n;i++){
vector<double> temp(5*(i+1)+1,0);
for(int j = 0;j<dp.size();j++){
for(int k = 0;k<6;++k){
temp[j+k] += dp[j] * (1.0/6.0);
}
}
dp = temp;
}
return dp;
}
};

结果

1
2
3
4
5
6
7
8
9
10
11
12
执行用时:
0 ms
, 在所有 C++ 提交中击败了
100.00%
的用户
内存消耗:
6.1 MB
, 在所有 C++ 提交中击败了
79.26%
的用户
通过测试用例:
11 / 11