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

题目

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

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

字符串中 0 的数量最多为 k。
字符串中 1 的数量最多为 k。
返回一个整数，表示 s 的所有满足 k 约束 的
子字符串
的数量。

示例 1：
输入：s = “10101”, k = 1
输出：12
解释：
s 的所有子字符串中，除了 “1010”、“10101” 和 “0101” 外，其余子字符串都满足 k 约束。

示例 2：
输入：s = “1010101”, k = 2
输出：25
解释：
s 的所有子字符串中，除了长度大于 5 的子字符串外，其余子字符串都满足 k 约束。

示例 3：
输入：s = “11111”, k = 1
输出：15
解释：
s 的所有子字符串都满足 k 约束。

提示：

1 <= s.length <= 50
1 <= k <= s.length
s[i] 是 ‘0’ 或 ‘1’。

思路

可以使用枚举，两层的循环遍历每一种可能的组合，把 1 和 0 是个数用一个数组记录（注意每遍历一种可能，重置数组）。

代码

class Solution {
public:
    int countKConstraintSubstrings(string s, int k) {
        int n = s.size(), res = 0;
        for (int i = 0; i < n; ++i) {
            int count[2] = {0};
            for (int j = i; j < n; ++j) {
                count[s[j] - '0']++;
                if (count[0] > k && count[1] > k) {
                    break;//这个组合不行，那么以这个为基础的增加组合也不行
                }
                res++;
            }
        }
        return res;
    }
};

总结

• 在 count[s[j] - ‘0’]++; 这种用法通常出现在统计字符串中各个数字出现的频率