Leetcode2466-统计构造好字符串的方案数

原创已于 2022-11-15 16:57:58 修改 · 201 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode #算法 #动态规划

于 2022-11-13 16:08:39 首次发布

Leetcode 同时被 2 个专栏收录

130 篇文章

订阅专栏

4 篇文章

订阅专栏

定义状态f[i]为能构造长度为i的字符串的种类，则状态转移方程为：

$gif.latex?f%5Bi%5D%3Df%5Bi-zero%5D+f%5Bi-one%5D$

依次递推到长度大于等于high为止，答案为low到high之间的总和。

class Solution {
public:
    const int mod=1e9+7;
    long long ans=0;
    int countGoodStrings(int low, int high, int zero, int one) {
        int dp[high+1];
        memset(dp,0,sizeof dp);
        dp[0]=1;
        for(int i=0;i<=high;i++){
            if(i>=zero) dp[i]=dp[i-zero];
            if(i>=one) dp[i]=(1LL*dp[i]+dp[i-one])%mod;
            if(i>=low) ans=(ans+1LL*dp[i])%mod;
        }
        return ans;
    }
};

时间复杂度：O（n）

空间复杂度：O（n）