nozomi和字符串+尺取法

题目描述：

nozomi看到eli在字符串的“花园”里迷路了，决定也去研究字符串问题。
她想到了这样一个问题：
对于一个 “01”串而言，每次操作可以把 0字符改为1字符，或者把 1 字符改为 0字符。所谓“01”串，即只含字符 0 和字符 1 的字符串。
nozomi有最多k次操作的机会。她想在操作之后找出一个尽可能长的连续子串，这个子串上的所有字符都相同。
nozomi想问问聪明的你，这个子串的长度最大值是多少？
注：k 次操作机会可以不全部用完。
如果想知道连续子串的说明，可以去问问eli，nozomi不想再讲一遍。

输入描述：

第一行输入两个正整数n和k（ 1 <= k<=n<=200000)输入仅有一行，为一个长度为n的、仅由字符 0 和 1 组成的字符串。

输出描述：

一个正整数，为满足条件的子串长度最大值。

示例：
输入

5 1
10101

输出

3

思路：

自然0 -> 1 和 1 -> 0 取最大值即可
尺取法，用 l , r 取维护一段区间内是否满足条件，取最大区间即可。用sum 去记录区间内需要替换字符数，如果该区间sum <= k 则可以继续 r ++ ,如果碰到需要替换的字符 sum ++, 如果 sum > k,先判断 str[l] 是否为需要替换字符，如果是sum --，否则 sum 不变，再 l ++，如此进行直至 r == n 结束。

代码：

#include<bits/stdc++.h>
using namespace std;
int n , k;
string str;
int change(char a, char b)// a 替换为 b
{
   int sum = 0;
   if(str[0] == a)
       sum = 1;
   int l = 0 , r = 0;
    int ans = 0;
   while(r < n)
   {
      if(sum <= k)
      {
          ans = max(ans, r - l + 1);
          r ++;
          if(str[r] == a)
           sum ++;
      }
      else
      {
          if(str[l] == a)
             sum --;
          l ++;
      }
   }
   return ans;
}
int main()
{
   cin >> n >> k >> str;
   cout << max(change('0','1'), change('1', '0')) << endl;
   return 0;
}