【Leetcode】2533. Number of Good Binary Strings

本文链接：https://blog.csdn.net/qq_46105170/article/details/141291415

题目地址：

https://leetcode.com/problems/number-of-good-binary-strings/description/

给定两个长度 $1\le l_1\le l_2$ ，和两个非负整数 $o, z$ 。现在要构造 $01$ 字符串 $s$ 使得 $s$ 的 $1$ 子串（ $1$ 子串意思是只含 $1$ 且左右无法继续延伸了的子串）的长度都是 $o$ 的倍数，且使得 $s$ 的 $0$ 子串（同上）的长度都是 $z$ 的倍数。问这样的 $s$ 且长度处于 $l_1,l_2]$ 之间的有多少个。

设 $f [k]$ 表示长度为 $k$ 的满足条件的串有多少个，对于任何长度为 $k$ 的满足条件的串可以分为两类，一类以 $1$ 结尾，如果如此，那么去掉结尾的 $o$ 个 $1$ ，剩余的串肯定也满足条件，个数是 $f [k - o]$ 个；同理以 $0$ 结尾的满足条件的有 $f [k - z]$ 个，从而 $f [k] = f [k - o] + f [k - z]$ 代码如下：

class Solution {
 public:
  int goodBinaryStrings(int minL, int maxL, int one, int zero) {
    constexpr int MOD = 1e9 + 7;
    vector<int> f(maxL + 1, 0);
    f[0] = 1;
    int res = 0;
    for (int i = 1; i <= maxL; i++) {
      if (i >= one) f[i] = (f[i] + f[i - one]) % MOD;
      if (i >= zero) f[i] = (f[i] + f[i - zero]) % MOD;
      if (i >= minL) res = (res + f[i]) % MOD;
    }
    return res;
  }
};

时空复杂度 $O(l_2)$ 。