题目内容
把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]
|
限制:
解法一:动态规划
// 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
|