2020牛客寒假算法基础集训营1 H nozomi和字符串

题目描述

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

输入描述:

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

输出描述:

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

示例1

输入

5 1
10101

输出

3

滑动窗口

#include<bits/stdc++.h>
#define GET_POS(c,x) (lower_bound(c.begin(),c.end(),x)-c.begin())
#define CASET int ___T; scanf("%d", &___T); for(int cs=1;cs<=___T;cs++)
#define MP make_pair
#define PB push_back
#define MS0(X) memset((X), 0, sizeof((X)))
#define MS1(X) memset((X), -1, sizeof((X)))
#define LEN(X) strlen(X)
#define F first
#define S second
#define ll long long
using namespace std;
typedef long long LL;
typedef unsigned long long ULL;
typedef long double LD;
const int mod = 1e8 + 7;
const int N = 1e6 + 10;
const int INF = 0x3f3f3f3f;
int solve(string s,int k)
{
	int count[26]={0};
	int l,r,res,maxcount; l=r=res=maxcount=0;
	while(r<s.size())
	{
		count[s[r]-'0']++;
		maxcount=max(maxcount,count[s[r]-'0']);
		while(r-l+1-maxcount>k)
		{
			count[s[l]-'0']--;
			l++;
		}
		res=max(res,r-l+1);
		
		r++;
	}
	return res;
}
int main()
{
	string str;
	int k,n;
	cin>>n>>k;
	cin>>str;
	cout<<solve(str,k)<<endl;
 }