leetcode-剑指 Offer 60. n个骰子的点数

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

示例 1:
输入: 1
输出: [0.16667,0.16667,0.16667,0.16667,0.16667,0.16667]

限制：

1 <= n <= 11
解题思路：
二维动态规划 ：dp[n] [j] 表示投掷n枚骰子，出现点数和为j的次数。
动态方程：dp[i] [j]+=dp[i-1[j-k] (k的取值范围是1-6）,也就是当前点数是在前一个骰子的基础上+点数(1-6)
某点数和出现的概率=该点数出现的次数/出现的总次数
比如2个骰子，考虑和为5出现的情况（ 2+3、3+2）虽然是一样的，但是同样算两次，所以总次数为pow(6,n)
代码的注释部分比较详细

class Solution {
public:
    vector<double> dicesProbability(int n) {
        vector<double> ans(5*n+1,0);   //n个骰子的点数为n...6*n
        vector<vector<int>>dp(n+1,vector<int>(6*n+1,0));   //骰子范围为0-n,点数的范围是0-6*n
        int m=n+1,l=6*n+1;
        for(int i=1;i<=6;i++)  //初始化一个骰子的时候，每个点数出现的次数一定是1
        dp[1][i]=1;
        for(int i=2;i<m;i++)
        {
            for(int j=i;j<l;j++)
            {
                for(int k=1;k<=6;k++)   //在前一个骰子的基础上+点数(1-6)即为当前的点数和
                {
                    if(j-k>0)
                    dp[i][j]+=dp[i-1][j-k];
                    else break;
                }
            }
        }
        double sum=pow(6,n);
        int k=0;
        for(int i=n;i<=6*n;i++)
        ans[k++]=dp[n][i]/sum;  //每个和出现的概率为：该和出现的次数/一共可以出现的骰子次数
        return ans;

    }
};