剑指offer-面试题 60：骰子的点数

题目描述： 

将一个骰子投掷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]。

算法：循环迭代

class Solution {
public:
    vector<int> numberOfDice(int n) {
        vector<int>value(6*n+1);
        if(n<1)
            return value;
        for(int i=1; i<=6; i++)
            value[i] = 1;
        
        
        for(int i = 2; i<=n; i++)
        {
            vector<int>temp(6*n+1);
            for(int j=i; j<=6*i; j++)
            {
                for(int k=1; k<=6 && k<=j; k++)
                {
                    temp[j] += value[j-k];
                }
            }
            value = temp;
        }
        vector<int>result;
        if(n==1)
        {
            for(int i=0; i<6; i++)
                result.push_back(1);
            return result;
        }
        for(int i=n; i<=6*n; i++)
        {
            result.push_back(value[i]);
        }
        return result;
    }
};

算法：动态规划

class Solution {
public:
    vector<int> numberOfDice(int n) {
        vector<int>value;
        if(n<1)
            return value;
        for(int i=n;i<=6*n;i++)
        {
            value.push_back(getNum(n, i));
        }
        return value;
    }
    int getNum(int n, int i)
    {
        if(i<n || i>6*n || n<1)
            return 0;
        if(n==1)
            return 1;
        int count = getNum(n-1, i-1)+getNum(n-1, i-2)+getNum(n-1, i-3)+getNum(n-1, i-4)+getNum(n-1, i-5)+getNum(n-1, i-6);
        return count;
    }
};

 

 参考文献：

1. 数据结构与算法学习笔记#剑指Offer65：n个骰子的点数和 + 详细解析（Java、C/C++）
2. n个骰子点数和及各自出现的概率