【算法题】2466. 统计构造好字符串的方案数

最新推荐文章于 2024-11-03 17:23:52 发布

程序员小2

最新推荐文章于 2024-11-03 17:23:52 发布

阅读量166

点赞数

分类专栏： leetcode-数组文章标签：动态规划算法

本文链接：https://blog.csdn.net/kangbin825/article/details/129871054

版权

leetcode-数组专栏收录该内容

204 篇文章 1 订阅

订阅专栏

该问题是一个关于计算在给定限制内的好字符串数量的编程问题。使用动态规划方法，通过维护一个数组dp来跟踪不同长度的字符串，然后根据zero和one的值更新dp，最终计算出符合条件的字符串总数，并对结果进行模运算处理。

摘要由CSDN通过智能技术生成

题目：

给你整数 zero ，one ，low 和 high ，我们从空字符串开始构造一个字符串，每一步执行下面操作中的一种：

将 ‘0’ 在字符串末尾添加 zero 次。
将 ‘1’ 在字符串末尾添加 one 次。
以上操作可以执行任意次。

如果通过以上过程得到一个长度在 low 和 high 之间（包含上下边界）的字符串，那么这个字符串我们称为好字符串。

请你返回满足以上要求的不同好字符串数目。由于答案可能很大，请将结果对 109 + 7 取余后返回。

示例 1：

输入：low = 3, high = 3, zero = 1, one = 1
输出：8
解释：
一个可能的好字符串是 “011” 。
可以这样构造得到：“” -> “0” -> “01” -> “011” 。
从 “000” 到 “111” 之间所有的二进制字符串都是好字符串。
示例 2：

输入：low = 2, high = 3, zero = 1, one = 2
输出：5
解释：好字符串为 “00” ，“11” ，“000” ，“110” 和 “011” 。

提示：

1 <= low <= high <= 10^5
1 <= zero, one <= low

java代码：

class Solution {
    long MOD = 1000000007;
    public int countGoodStrings(int low, int high, int zero, int one) {
        long ans = 0;
        int[] dp = new int[high+1];
        dp[0] = 1;
        for(int i = 0;i<high+1;++i){
            //类似于爬楼梯 爬楼梯的不是规定是1步或者2步 这里是zero和one
            //走到这一步无非是两种状态 
            //上一步走  1.zero步 2.one步
            //所以这次的组合 是 前zero步的次数 + 前one步的次数
            if(i>=zero) dp[i] += dp[i-zero];
            if(i>=one)  dp[i] += dp[i-one];
            dp[i] %= MOD;
            if(i>= low) ans += dp[i];
            ans %= MOD;
        }
        return (int)ans;
    }
}