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

wobushihaorenF

已于 2022-11-15 16:57:58 修改

阅读量147

点赞数

分类专栏： Leetcode DP 文章标签：动态规划字符串构造状态转移方程时间复杂度空间复杂度

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

本文链接：https://blog.csdn.net/aaa7888210/article/details/127833364

版权

Leetcode 同时被 2 个专栏收录

130 篇文章 1 订阅

订阅专栏

4 篇文章 0 订阅

订阅专栏

定义状态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）