LeetCode每日一题3261---统计满足 K 约束的子字符串数量 II

一、题目描述

给你一个 二进制 字符串 s 和一个整数 k。

另给你一个二维整数数组 queries ，其中 queries[i] = [li, ri] 。

如果一个 二进制字符串 满足以下任一条件，则认为该字符串满足 k 约束：

字符串中 0 的数量最多为 k。
字符串中 1 的数量最多为 k。
返回一个整数数组 answer ，其中 answer[i] 表示 s[li..ri] 中满足 k 约束 的 子字符串的数量。

示例 1：
输入：s = “0001111”, k = 2, queries = [[0,6]]

输出：[26]

解释：
对于查询 [0, 6]， s[0…6] = “0001111” 的所有子字符串中，除 s[0…5] = “000111” 和 s[0…6] = “0001111” 外，其余子字符串都满足 k 约束。

示例 2：
输入：s = “010101”, k = 1, queries = [[0,5],[1,4],[2,3]]

输出：[15,9,3]

解释：
s 的所有子字符串中，长度大于 3 的子字符串都不满足 k 约束。

提示：
1 <= s.length <= 105
s[i] 是 ‘0’ 或 ‘1’
1 <= k <= s.length
1 <= queries.length <= 105
queries[i] == [li, ri]
0 <= li <= ri < s.length
所有查询互不相同

二、解题思路

这道题目与统计满足 K 约束的子字符串数量 I相似，就是需要处理多个子串，完全可以利用上一次的函数进行求解，但是由于这次的字符串长度太大，会导致超出时间限制，因此需要特殊的方法处理，可以使用双指针 + 前缀和 + 二分

三、代码

1、普通方法（超出时间限制）

class Solution {
   
public:
    // 计算符合条件的子串个数
    int countKConstraintSubstrings1(string& s, int k) {
   
        int l = 0, count = 0, one = 0, zero = 0;
        for (int i = 0; i < s.size(); i++) {
   
            if (s[i] == '1') one++;
            else zero++;

            // Shrink the window until both one and zero counts are <= k
            while (one >