骰子的点数

好久没有写算法题了，今天心血来潮看到一道dp，遂写了一下，发现写起来有点困难，果然算法要持续练啊！ 题目来源：剑指offer.

题曰：

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

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

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

数据范围

1≤n≤10

样例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]。

 暴力做法：TLE了，不细说了

class Solution {
public:
    vector<int> numberOfDice(int n) {
        vector<int> res;
        for(int i = n; i <= n * 6; i ++) res.push_back(dfs(n, i));
        return res;
    }

    int dfs(int n, int sum)
    {
        if(sum < 0) return 0;
        if(!n) return !sum;

        int ans = 0;
        for(int i = 1; i <= 6; i ++)
            ans += dfs(n - 1, sum - i);

        return ans;
    }
};

      

线性DP

class Solution {
public:
    vector<int> numberOfDice(int n) {
        vector<int> ans;
        int f[n + 1][n * 6 + 1];         //f表示前i次投掷，总和为j的集合， 属性为count
        memset(f, 0, sizeof f);
        for (int i = 1; i <= 6; i++) f[1][i] = 1;
        for (int i = 2; i <= n; i++ ) 
            for (int j = i; j <= 6 * n; j++)
                for (int k = 1; k <= 6; k++)        //表示最后一次投掷的点数
                    if (j > k)
                        f[i][j] += f[i - 1][j - k];
                        
        for (int i = n; i <= 6 * n; i ++) ans.push_back(f[n][i]);
        return ans;
    }               
};