leetcode - 1223. 掷骰子模拟

1223. 掷骰子模拟

有一个骰子模拟器会每次投掷的时候生成一个 $1$ 到 $6$ 的随机数。

不过我们在使用它时有个约束，就是使得投掷骰子时，连续 掷出数字 $i$ 的次数不能超过 $rollMax[i]$（$i$ 从 $1$ 开始编号）。

现在，给你一个整数数组 $rollMax$ 和一个整数 $n$，请你来计算掷 $n$ 次骰子可得到的不同点数序列的数量。

假如两个序列中至少存在一个元素不同，就认为这两个序列是不同的。由于答案可能很大，所以请返回 模 $10^9+7$ 之后的结果。

示例 1：

输入：n = 2, rollMax = [1,1,2,2,2,3]
输出：34
解释：我们掷 2 次骰子，如果没有约束的话，共有 6 * 6 = 36 种可能的组合。但是根据 rollMax 数组，数字 1 和 2 最多连续出现一次，所以不会出现序列 (1,1) 和 (2,2)。因此，最终答案是 36-2 = 34。

示例 2：

输入：$n=2,rollMax=[1,1,1,1,1,1]$
输出：$30$

示例 3：

输入：$n=3,rollMax=[1,1,1,2,2,3]$
输出：$181$

提示：

• $1<=n<=5000$
• $rollMax.length==6$
• $1<=rollMax[i]<=15$

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/dice-roll-simulation
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

---

解题思路

使用动态规划，使用三维数组存储状态变量；$dp[i][j][k]$表示第 $i$次掷骰子掷出数字 $j$ 时， $j$ 连续出现 $k$ 次的组合数量;

根据状态变量，可以得到状态转移方程：

• 当第$i$次掷骰子掷出数字$j$，$j$连续出现$k(k>1)$次的的组合数量为：
  $dp[i][j][k]=dp[i-1][j][k-1]$
• 当第$i$次掷骰子掷出数字$j$，$j$连续出现$k(k==1)$次的的组合数量为：
  $dp[i][j][1]=sum(dp[i-1][!j][:])$（其中$!j$表示在第$i-1$次掷骰子的数字是除了$j$以外的任何数字，$[:]$是考虑所有可能的情况）

根据状态转移方程和状态变量，使用C++进行编程，可以得到：

class Solution {
public:
    int dieSimulator(int n, vector<int>& roll) {
        int big = 0;  //计算roll中的最大值
        for(int i=0;i<6;++i)
        {
            big = max(big,roll[i]);
        }
        vector<vector<vector<int>>> dp(n+1,vector<vector<int>>(7,vector<int>(big+1, 0))); //状态变量dp[i][j][k]
        for(int i=1;i<=6;++i)   //初始化，第一次掷骰子的结果
        {
            dp[1][i][1] = 1;
        }
        for(int i=2;i<=n;++i)  //从第二次掷骰子开始使用状态转移方程
        {
            for(int j=1;j<=6;j++)  //每一次掷骰子有6中情况
            {
                for(int k=2;k<=roll[j-1];++k)  //当k的值不为1，即投掷骰子得到的数字是连续k次时，其状态转移方程如下：
                {
                    dp[i][j][k] = dp[i-1][j][k-1];
                }
                for(int k=1;k<=6;++k)   //当k的值为1，即投掷骰子的数字和上一次的结果是不连续的，其状态转移方程如下：
                {
                    if(k==j)  //忽略和上一次投掷结果一样的情况
                        continue;
                    for(int l=1;l<=big;++l)  //考虑和上一次投掷结果不一样的情况，对于上一次投掷，其有很多种情况，需要考虑所有的情况
                        dp[i][j][1] = (dp[i-1][k][l]+dp[i][j][1])%1000000007; 
                }
            }
        }
        int sum=0;  //计算所有可能的情况并返回
        for(int i=1;i<=6;++i)
        {
            for(int j=1;j<=roll[i-1];j++)
                sum = (sum + dp[n][i][j])%1000000007;
        }
        return sum;
    }
};