力扣 中等 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;
    }
}
相关推荐
jonyleek9 分钟前
数据可视化:JVS-BI仪表盘图表样式配置全攻略,打造个性化数据展示!
java·大数据·信息可视化·数据挖掘·数据分析·自动化·软件需求
WangMing_X10 分钟前
C# 单个函数实现各进制数间转换
java·开发语言·算法·c#·winform·软件
南宫生23 分钟前
贪心算法理论基础和习题【算法学习day.17】
java·学习·算法·leetcode·链表·贪心算法
海绵波波10724 分钟前
240. 搜索二维矩阵 II
数据结构·算法·矩阵
jc0803kevin29 分钟前
solidity的struct对象,web3j java解析输出参数
java·web3·solidity
Death20031 分钟前
PCL库中的算法封装详解
c++·qt·算法·3d
勇敢滴勇32 分钟前
【C++】继承和多态常见的面试问题
java·c++·面试
nice6666037 分钟前
初识JDBC
java·数据库·sql·mysql·idea
weixin_3077791337 分钟前
复变函数解题技巧
算法
计算机学姐39 分钟前
基于SpringBoot的汽车票网上预订系统
java·vue.js·spring boot·后端·mysql·java-ee·mybatis