力扣 中等 2466.统计构造好字符串的方案数

文章目录

题目介绍

题解

题意 :每次可以爬 zeroone 个台阶,返回爬 lowhigh 个台阶的方案数。

和上题337.组合总和 (链接)的思路一样,只不过是将可以爬的台阶数从数组换成了两个数(zero和one)

代码如下:

java 复制代码
class Solution {
    public int countGoodStrings(int low, int high, int zero, int one) {
        final int MOD = 1000000007;
        int ans = 0;
        int[] dp = new int[high + 1]; // dp[i] 表示构造长为 i 的字符串的方案数
        dp[0] = 1; // 构造空串的方案数为 1
        for (int i = 1; i <= high; i++) {
            if (zero <= i) dp[i] = dp[i - zero] % MOD;
            if (one <= i)  dp[i] = (dp[i] + dp[i - one]) % MOD;
        }
        for (int i = low; i <= high; i++) {
            ans = (ans + dp[i]) % MOD;
        }
        return ans;
    }
}

简化代码:将两个for循环合并成一个

java 复制代码
class Solution {
    public int countGoodStrings(int low, int high, int zero, int one) {
        final int MOD = 1000000007;
        int ans = 0;
        int[] dp = new int[high + 1]; // dp[i] 表示构造长为 i 的字符串的方案数
        dp[0] = 1; // 构造空串的方案数为 1
        for (int i = 1; i <= high; i++) {
            if (i >= zero) dp[i] = dp[i - zero];
            if (i >= one)  dp[i] = (dp[i] + dp[i - one]) % MOD;
            if (i >= low)  ans = (ans + dp[i]) % MOD;
        }
        return ans;
    }
}
相关推荐
DKPT6 分钟前
Eclipse,MyEclipse,IDEA,Vscode这些编译器和JDK的相爱相杀
java·eclipse·编辑器·intellij-idea·myeclipse
肠胃炎9 分钟前
真题246—矩阵计数
java·线性代数·算法·矩阵·深度优先
什码情况10 分钟前
微服务集成测试 -华为OD机试真题(A卷、JavaScript)
javascript·数据结构·算法·华为od·机试
前行的小黑炭1 小时前
设计模式:为什么使用模板设计模式(不相同的步骤进行抽取,使用不同的子类实现)减少重复代码,让代码更好维护。
android·java·kotlin
Java技术小馆1 小时前
如何设计一个本地缓存
java·面试·架构
XuanXu2 小时前
Java AQS原理以及应用
java
罗西的思考2 小时前
[2W字长文] 探秘Transformer系列之(23)--- 长度外推
人工智能·算法
风象南5 小时前
SpringBoot中6种自定义starter开发方法
java·spring boot·后端
mghio13 小时前
Dubbo 中的集群容错
java·微服务·dubbo