3258. 统计满足 K 约束的子字符串数量 I

高效计算满足 K 约束的子字符串数量：两种解法解析

在字符串处理问题中，如何高效地计算满足特定条件的子字符串个数是一个常见的挑战。在本篇博客中，我们将探讨一个典型的字符串问题：给定一个二进制字符串 s 和一个整数 k，要求我们计算所有满足如下条件的子字符串数量：

• 子字符串中 0 的数量不超过 k，或者
• 子字符串中 1 的数量不超过 k

我们将展示两种不同的解法：一种是暴力解法，另一种是使用 滑动窗口 技术的优化解法。

问题分析

假设我们给定了一个二进制字符串 s 和整数 k，我们的目标是计算字符串中所有满足以下条件的子字符串数量：

• 子字符串中 0 的数量最多为 k
• 子字符串中 1 的数量最多为 k

例如：

示例 1：

输入: s = "10101", k = 1
输出: 12

示例 2：

输入: s = "1010101", k = 2
输出: 25

方案 1：暴力解法

暴力解法的思路很简单：枚举字符串中的所有子字符串，并检查每个子字符串中 0 和 1 的数量是否满足条件。如果满足条件，就计数。

代码实现：

class Solution:
    def countKConstraintSubstrings(self, s: str, k: int) -> int:
        n = len(s)
        ans = 0
        for i in range(n):
            cnt0 = 0
            cnt1 = 0