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

1. 题目链接

2. 解题思路

   • 小字符串不满足了，那么包含此小字符串的大字符串肯定不满足。
   • 大字符串满足，那么该字符串的所有子串都满足条件。所以，我们就可以把问题转成，「以i开头的字符串，满足条件的，最长多长？」然后再根据这个结果，计算以i开头的所有子串数目即可。
   • 计算了「以i开头的最长的结果」，怎么计算「以i+1开头的最长的结果」？滑动窗口，假设以i开头结果是[i, j]，窗口大小就是[i, j]，此时，把i移出窗口，看右侧窗口是否还能扩大。

3. 代码

   class Solution {
   public:int countKConstraintSubstrings(string s, int k) {int j = -1;int ans = 0;int n = s.length();int oneNums = 0;int zeroNums = 0;for (int i = 0; i < n; ++i) {// 把 i - 1移出窗口if (i > 0) {switch (s[i - 1]) {case '0':zeroNums--;break;case '1':oneNums--;break;}}// 右侧窗口能扩吗？while(j + 1 < n) {if (s[j + 1] == '0') {if (zeroNums < k || oneNums <= k) {j++;zeroNums++;    } else {break;}} else if (s[j + 1] == '1' ) {if (oneNums < k || zeroNums <= k) {j++;oneNums++;    } else {break;}} else {break;}}// 计算结果ans += j - i + 1;}return ans;}
   };