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

最新推荐文章于 2024-10-14 12:54:09 发布

咔咔咔的

最新推荐文章于 2024-10-14 12:54:09 发布

阅读量418

点赞数 4

分类专栏： leetcode 文章标签： c++

本文链接：https://blog.csdn.net/weixin_45256307/article/details/141102182

版权

leetcode 专栏收录该内容

593 篇文章 2 订阅

订阅专栏

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

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

代码如下：

class Solution 
{
public:
    int numberOfStableArrays(int zero, int one, int limit) 
    {
        vector<vector<array<int, 2>>> f(zero + 1, vector<array<int, 2>>(one + 1));
        for (int i = 1; i <= min(limit, zero); i++)
        {
            f[i][0][0] = 1;
        }

        for (int j = 1; j <= min(limit, one); j++)
        {
            f[0][j][1] = 1;
        }

        for (int i = 1; i <= zero; i++)
        {
            for (int j = 1; j <= one; j++)
            {
                f[i][j][0] = ((long long)f[i - 1][j][0] + f[i - 1][j][1] + (i > limit ? mod - f[i - limit - 1][j][1] : 0)) % mod;
                f[i][j][1] = ((long long)f[i][j - 1][0] + f[i][j - 1][1] + (j > limit ? mod - f[i][j - limit - 1][0] : 0)) % mod;
            }
        }

        return (f[zero][one][0] + f[zero][one][1]) % mod;    
    }
private:
	const int mod = 1'000'000'007;
};