LeetCode 3130.找出所有稳定的二进制数组II

题目描述

原题链接：LeetCode 3130.找出所有稳定的二进制数组II

解题思路

动态规划（$o(n^2)$）

定义dp0[i][j]为以$0$结尾的有着$i$个$0$和$j$个$1$的稳定的二进制数组的数目，定义dp1[i][j]为以$1$结尾的有着$i$个$0$和$j$个$1$的稳定二进制数组的数目。
对于dp0[i][j]的更新，我们可以先进行分类讨论：

1. 当$i\le min(limit,zero),j=0$时，有且仅有一种方案，就是全为$1$。
2. 当$j\ne 0$时，又可以分为两种情况：
   1. 当$i\le limit$时，dp0[i][j]可以由dp0[i-1][j]和dp1[i-1][j]更新而来，dp0[i][j]=dp0[i-1][j]+dp1[i-1][j]。
   2. 当$i>limit$时，dp0[i-1][j]和dp1[i-1][j]中只有一种情况不满足添加$1$个$0$之后，连续$0$的数量不大于$limit$，就是当后面$limit$个数都是$0$的时候。这时候的方案数与dp1[i-1-limit][j]相同。这里是dp0而不是dp1是因为假如是dp0，那么也就是说这个数组有不止$limit$个$0$，也就不满足题意。这个时候的状态转移方程为：dp0[i][j]=dp0[i-1][j]+dp1[i-1][j]-dp1[i-1-limit][j]。
3. 当$i=0$时，不存在满足题意的方案。dp0[i][j]=0。

同理也可以得到dp1[i][j]的状态转移方程。总的状态转移方程如下：
$$dp0[i][j]=\{\begin{array}{ll}1,& ifi\le min(limit,zero)andj=0\\ dp0[i-1][j]+dp1[i-1][j],& ifi\le limitandj\ne 0\\ dp0[i-1][j]+dp1[i-1][j]-dp1[i-1-limit][j],& ifi>limitandj\ne 0\\ 0,& ifi=0\end{array}$$
$$dp1[i][j]=\{\begin{array}{ll}1,& ifj\le min(limit,one)andi=0\\ dp0[i][j-1]+dp1[i][j-1],& ifj\le limitandi\ne 0\\ dp0[i][j-1]+dp1[i][j-1]-dp0[i][j-1-limit],& ifj>limitandi\ne 0\\ 0,& ifj=0\end{array}$$

###C++代码

class Solution {
public:
    typedef long long LL;
    int numberOfStableArrays(int zero, int one, int limit) {
        LL dp1[1010][1010] = {0};
        LL dp0[1010][1010] = {0};
        LL mod = 1e9+7;
        for(int i = 1; i <= min(limit, zero); i++) {
            dp0[i][0] = 1;
        }
        for(int j = 1; j <= min(limit, one); j++) {
            dp1[0][j] = 1;
        }
        for(int i = 1; i <= zero; i++) {
            for(int j = 1; j <= one; j++) {
                if(i <= limit) {
                    dp0[i][j] = dp0[i-1][j]+dp1[i-1][j];
                } else {
                    dp0[i][j] = dp0[i-1][j]+dp1[i-1][j]-dp1[i-1-limit][j];
                }
                dp0[i][j] = (dp0[i][j] % mod+mod)%mod;

                if(j <= limit) {
                    dp1[i][j] = dp0[i][j-1]+dp1[i][j-1];
                } else {
                    dp1[i][j] = dp0[i][j-1]+dp1[i][j-1]-dp0[i][j-1-limit];
                }
                dp1[i][j] = (dp1[i][j] % mod+mod)%mod;
            }
        }
        return (dp0[zero][one]+dp1[zero][one])%mod;
    }
};

注意： 在对结果进行取余的时候要注意防止出现负数，因此要先加上一个$10^9+7$再取余。如果不那么做，那么在dp0[i][j] = dp0[i-1][j]+dp1[i-1][j]-dp1[i-1-limit][j];的时候，由于dp0[i-1][j]+dp1[i-1][j]虽然比dp1[i-1-limit][j]大，但是由于前一项取余之后可能很小，后一项取余之后也可能很大，因此相减可能出现负数，如果不处理可能导致最终的结果为负。