牛客 2019年湘潭大学程序设计竞赛 F:Black & White（尺取）

(思路：尺取，先把连续的“0”看作一块，忽略“1”，记下每一块的左端点和右端点，尺取求一下最大值；再把连续的“1”看作一块，忽略“0”，记下每一块的左端点和右端点，尺取求一下最大值；再对两个最大值取一下最大值，然后再和n取一下较小值，因为我的算法有可能求出的值大于n。详情看注释。)

 

#include <bits/stdc++.h>
using namespace std;
const int N=1e5+10;
struct node
{
	int l,r;
}a[N];
char s[N];
int main(void)
{
	int t;
	cin>>t;
	while(t--)
	{
		int n,m;
		scanf("%d%d%s",&n,&m,s+1);
		int cnt=0,i;
		int ans;
		int ll,rr,mm=m;
		//把"0"分为小块 
		i=1;
		while(i<=n)
		{
			if(s[i]=='0')
			{
				int j=i;
				while(j<=n&&s[j]=='0')
					j++;
				a[cnt].l=i;
				a[cnt++].r=j-1;
				i=j;				
			}
			else
				i++;
		}
		a[cnt++].l=2*N;
		ans=0;
		mm=m;
		ll=0;
		rr=1;
		while(rr<cnt)
		{
			while(rr<cnt&&mm>=0)
			{
				ans=max(ans,a[rr-1].r-a[ll].l+1+mm);
				//每次都算rr-1这一块的r右端点和ll这一块的l左端点的差值  
				//这里加mm的意思是指取rr-1和rr这两块的间隔中的一部分 
				//如果取到了最后最后一块，那么取得就是ll-1和ll这两块的间隔的一部分 
				//这也是最后要和n取较小值的原因 
				mm-=(a[rr].l-a[rr-1].r-1);
				//减去这一块和下一块的间隔 
				rr++;
			}
			//尺取，不取ll这一块 
			mm+=a[ll+1].l-a[ll].r-1;		
			ll++;
		}		
		
		i=1;
		cnt=0;
		while(i<=n)
		{
			if(s[i]=='1')
			{
				int j=i;
				while(j<=n&&s[j]=='1')
					j++;
				a[cnt].l=i;
				a[cnt++].r=j-1;
				i=j;				
			}
			else
				i++;
		}
		a[cnt++].l=2*N;
		mm=m;
		ll=0;
		rr=1;
		while(rr<cnt)
		{
			while(rr<cnt&&mm>=0)
			{
				ans=max(ans,a[rr-1].r-a[ll].l+1+mm);	
				mm-=(a[rr].l-a[rr-1].r-1);
				rr++;
			}	
			mm+=a[ll+1].l-a[ll].r-1;		
			ll++;
		}
		cout<<min(ans,n)<<endl;
	}

	return 0;
 }