剑指 Offer 60. n个骰子的点数C++

题目描述：把n个骰子扔在地上，所有骰子朝上一面的点数之和为s。输入n，打印出s的所有可能的值出现的概率。你需要用一个浮点数数组返回答案，其中第 i 个元素代表这 n 个骰子所能掷出的点数集合中第 i 小的那个的概率。

解题思路：n 枚骰子掷出的点数的范围是 [n, 6n]，也就是最后数组的长度为 5n + 1，因为每个骰子掷出每个数字的概率都是 1/6，所以对于 n 枚骰子，每次掷出的数字组合的概率都为 (1/6)^n。也就是说如果和为 s 的组合有 k 种，那么和为 s 的概率为 k(1/6)^n我们这里采用动态规划的思想进行处理，假设我们我们投掷完 n 枚骰子后总点数为 j，使用 dp[n][j] 表示投完 n 枚骰子后总点数为 j 的出现次数，那么我们考虑递推关系式，投第 n 枚骰子可以由投第 n-1 枚骰子转换而来，也就是如果投第 n-1 枚骰子后总点数为 j-i (1 ≤ i ≤ 6)，那么第 n 次掷出 i 即可满足条件。

代码：

class Solution {
public:
    vector<double> twoSum(int n) {
        vector<vector<double>>dp(n+1,vector<double>(6*n+1,0));
        vector<double>ans;
        for(int i=1;i<=n;i++)
          for(int j=i;j<=6*i;j++){
              if(i==1){
                  dp[i][j]=1;
                  continue;
              }
              for(int k=1;k<=6;k++){
                  if(j-k>=i-1) dp[i][j]+=dp[i-1][j-k];
              }
          }
          for(int i=n;i<=6*n;i++){
              ans.push_back(dp[n][i]*pow(1.0/6,n));
          }
          return ans;
    }
};