计数包含给定字符K 次的子字符串

问题描述

给定一个字符串S和一个整数K，计算S中有多少个长度为K的子字符串包含给定的字符K次。

解决方案

我们可以使用滑动窗口算法来解决该问题。具体算法步骤如下：

1. 定义一个计数器，用于记录字符串中包含给定字符K的子字符串数目。
2. 定义一个长度为K的滑动窗口，从字符串的第一个字符开始依次向后滑动。
3. 对于每个滑动窗口，统计其中字符K的出现次数。如果该子字符串中字符K的出现次数等于给定的次数K，计数器加1。
4. 继续滑动窗口，直到扫描完整个字符串S为止。返回计数器的值即可。

代码实现

def count_substring(s: str, k: int) -> int:
    count = 0
    window = s[:k]
    k_count = window.count('K')
    if k_count == k:
        count += 1
    for i in range(k, len(s)):
        k_count += 1 if s[i] == 'K' else 0
        k_count -= 1 if s[i - k] == 'K' else 0
        window = s[i - k + 1:i + 1]
        if k_count == k:
            count += 1
    return count

时间复杂度

该算法的时间复杂度为O(n)，其中n为字符串S的长度。

空间复杂度

该算法的空间复杂度为O(k)，其中k为给定的字符K出现的次数和给定的次数K的较大值。