题目描述

题解思路

组成长度为i的字符串个数 = 组成长度为i - zero的字符串个数 + 组成长度为i - one的字符串个数
设数组f中i号元素的值为组成长度为i的字符串个数
则状态转移方程为f[i] = f[i - zero] + f[i - one]，其中需要舍弃为负数的下标
最终我们计算完f中所有值后，计算f数组中[low, high]区间和就为此题的答案
我们需要注意在进行计算过程中要对1000000007取余

题解代码

func countGoodStrings(low int, high int, zero int, one int) int {
    f := make([]int, high + 1)
    f[0] = 1
    for i := 1; i <= high; i++ {
        if i >= zero {
            f[i] = f[i - zero]
        }
        if i >= one {
            f[i] += f[i - one]
        }

        f[i] = f[i] % 1000000007
    }

    ans := 0
    for i := low; i <= high; i++ {
        ans += f[i]
    }

    return ans % 1000000007
}

题目链接

https://leetcode.cn/problems/count-ways-to-build-good-strings/