lintcode(18)——骰子求和

20. 骰子求和

扔 n 个骰子，向上面的数字之和为 S。给定 n，请列出所有可能的 S值及其相应的概率。

样例 1：

输入：n = 1
输出：[[1, 0.17], [2, 0.17], [3, 0.17], [4, 0.17], [5, 0.17], [6, 0.17]]
解释：掷一次骰子，向上的数字和可能为1,2,3,4,5,6，出现的概率均为 0.17。

样例 2：

输入：n = 2
输出：[[2,0.03],[3,0.06],[4,0.08],[5,0.11],[6,0.14],[7,0.17],[8,0.14],[9,0.11],[10,0.08],[11,0.06],[12,0.03]]
解释：掷两次骰子，向上的数字和可能在[2,12]，出现的概率是不同的。

注意事项

你不需要关心结果的准确性，我们会帮你输出结果。

思路：

       我们知道，如果只有一个骰子，那么它的可能区间是1到6，如果两个骰子，那么可能的区间是是2到12，如果是3个骰子，那么可能的区间是3到18，如果是n个骰子，那么可能的区间是n到6*n。
       我们创建一个数组，每一行代表每一种情况，第一行代表一个骰子的情况，第二行代表两个骰子的情况，第三行代表三个骰子的情况。

       我们假设有n个骰子，总的点数和为sum，那么在前面n-1个骰子的情况，最后一个可以有Sum-1、Sum-2、Sum-3、Sum-4、Sum-5、Sum-6的六种情况，而最后的那个骰子有下面的情况：

（n-1，sum-1）：第n个骰子扔出了1，等同n-1个骰子扔出了sum-1的情况。

（n-1，sum-2）：第n个骰子扔出了2，等同n-1个骰子扔出了sum-2的情况。

（n-1，sum-3）：第n个骰子扔出了3，等同n-1个骰子扔出了sum-3的情况。

（n-1，sum-4）：第n个骰子扔出了4，等同n-1个骰子扔出了sum-4的情况。

（n-1，sum-5）：第n个骰子扔出了5，等同n-1个骰子扔出了sum-5的情况。

（n-1，sum-6）：第n个骰子扔出了6，等同n-1个骰子扔出了sum-6的情况。

那么n个骰子扔出了sum的情况等于上面六种情况相加。

 n = 1时：f(1,1) = f(1,2) = f(1,3) = f(1,4) = f(1,5) = f(1,6) = 1
而 n = 2时：f(2,2) = f(1,1) = 1

f(2,3) = f(1,2) + f(1,1) = 2

...
f(2,6) = f(1,5) + f(1,4) + f(1,3) + f(1,2) + f(1,1)
f(2,7) = f(1,6) + f(1,5) + f(1,4) + f(1,3) + f(1,2) + f(1,1) = 6

表格如下：

 

class Solution {
public:
    vector<pair<int, double>> dicesSum(int n) {
        // Write your code here
        vector<pair<int, double>> results;
        vector<vector<double>> f(n+1,vector<double>(6*n+1));
        for(int i=1;i<=6;i++)   f[1][i]=1.0/6;
        for(int i=2;i<=n;i++)
        {
            for(int j=i;j<=6*i;j++)
            {
                for(int k=1;k<=6;k++)
                {
                    if(j>k)
                    {
                        f[i][j]+=f[i-1][j-k];
                    }
                }
                f[i][j]/=6.0;
            }
            
        }
        for(int i=n;i<=6*n;i++)
        {
            results.push_back(make_pair(i,f[n][i]));
        }
        return results;
    }
};