给你一个 二进制 字符串 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 约束。
思路
采用双循环的方法,i 用来代表每次检查字符串的起始位置,j 表示检查字符串的长度,每次 i 移动的时候新建一个 count 数组来记 i 到 j 的 0/1 个数,最后统计结果。
代码(C++)
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;}
};
代码(Python)
class Solution:def countKConstraintSubstrings(self, s: str, k: int) -> int:n = len(s)res = 0for i in range(n):count = [0, 0]for j in range(i, n):count[int(s[j])] += 1if count[0] > k and count[1] > k:breakres += 1return res