把n个骰子扔在地上,所有骰子朝上一面的点数之和为s。输入n,打印出s的所有可能的值出现的概率。
你需要用一个浮点数数组返回答案,其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。
思路:如果暴力找出所有的解,时间复杂度是6^n;
时间复杂度太高了,所以想要加速,然后想到了换硬币的题目, n个骰子,最大和为n6,f(i,j )为骰子是i个时,和为j 的出现次数。则会有:f (i , j) += f(i-1, j-(1到6))这样,就是i个骰子时,是依赖于i-1个骰子的,然后 j 的最大值应该是从i(每个骰子都是1)到n6 .
所以综上:一共有三行for,i 骰子数;j 和的大小,cur 1-6的值
代码:
class Solution {
public:
vector<double> twoSum(int n) {
//n最大是11,
int dp[15][70];
memset(dp,0, sizeof(dp));
for(int i=1;i<=6;i++){
dp[1][i]=1;
}
for(int i=2 ;i<=n; ++i){
for(int j=i; j<=6*i; ++j){ //对应的i是多少个骰子的最小值,3个骰子从3开始
for(int cur=1;cur<=6;cur++){
if(j-cur<0) //不能够为负数,和不可能负数
break;
dp[i][j]+=dp[i-1][j-cur];
}
}
}
int sum=pow(6,n);
vector<double> res;
for(int i=n;i<= 6*n;i++)
res.push_back(dp[n][i]*1.0/sum);
return res;
}
};