题目:
把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
输入: 1
输出: [0.16667, 0.16667, 0.16667, 0.16667, 0.16667, 0.16667]
输入: 2
输出: [0.02778, 0.05556, 0.08333, 0.11111, 0.13889,
0.16667, 0.13889, 0.11111, 0.08333, 0.05556, 0.02778]
解析:
动态规划
参考答案:
class Solution {
public:
vector<double> twoSum(int n) {
//n <= 11
vector<double>dp(6*n + 1, 0);
vector<double> ans;
for(int i = 1; i <= n; i ++){
for(int j = 6*i; j >= i; j --){
dp[j] = 0;
if(i == 1) {
dp[j] = 1;
continue;
}
for(int k = 1; k <= 6; k ++){
//需要保证硬币总和得大于等于硬币数
if(j - k >= i - 1) dp[j] += dp[j - k];
}
}
}
for(int i = n; i <= 6*n; i ++){
ans.push_back(dp[i] * pow(1.0/6, n));
}
return ans;
}
};