20190630-剑指offer60-AcWing-80. 骰子的点数

剑指offer60-AcWing-80. 骰子的点数

将一个骰子投掷n次，获得的总点数为s，s的可能范围为n~6n。

掷出某一点数，可能有多种掷法，例如投掷2次，掷出3点，共有[1,2],[2,1]两种掷法。

请求出投掷n次，掷出n~6n点分别有多少种掷法。

样例1

输入：n=1

输出：[1, 1, 1, 1, 1, 1]

解释：投掷1次，可能出现的点数为1-6，共计6种。每种点数都只有1种掷法。所以输出[1, 1, 1, 1, 1, 1]。

样例2

输入：n=2

输出：[1, 2, 3, 4, 5, 6, 5, 4, 3, 2, 1]

解释：投掷2次，可能出现的点数为2-12，共计11种。每种点数可能掷法数目分别为1,2,3,4,5,6,5,4,3,2,1。

      所以输出[1, 2, 3, 4, 5, 6, 5, 4, 3, 2, 1]。

思路：

    动态规划问题。设数组sumNode[i][j] 表示第i次投掷骰子，出现和为j的次数为多少。
    其中sumNode[i][j] += sumNode[i - k]，其中k为第i次投掷可能出现的点数。因此这个二维数组比较耗内存，因此可以转为滚动数组，把二维数组转为一维数组。

C++ code：

class Solution {
public:
    vector<int> numberOfDice(int n) {
        vector<int> sumNode(6 * n + 1, 0);
        //第一次骰子出现1-6的次数
        for(int i = 1; i <=6; i++){
            sumNode[i] = 1;
        }
        //i表示投掷的次数，或者是理解为骰子的个数
        for(int i = 2; i <= n; i++){
        	//j表示第i次投掷，i个骰子和为j的次数
            for(int j = 6 * i; j > 0; j--){
                sumNode[j] = 0;
                //第i次投掷出现的点数
                for(int k = 6; k >= 1; k--){
                    if(j - k < 0){
                        continue;
                    }
                    sumNode[j] += sumNode[j - k];
                }
            }
        }
        return vector<int>(sumNode.begin() + n, sumNode.end());
    }
};