ZZULI OJ 1895: 985的0-1串难题

1895: 985的0-1串难题

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 117   Solved: 21

Submit Status Web Board

Description

985有一个长度为n的0-1串，已知他最多可以修改k次（每次修改一个字符即0->1 或者 1->0），他想知道连续的全1子串最长是多少。

Input

第一行输入一个整数t，代表有t组测试数据。

每组数据第一行输入两个整数n，k分别代笔上面的信息。

注：1 <= t <= 12，1 <= n <= 100000，0 <= k <= 100000。

Output

一个整数代表可以得到的最大长度。

Sample Input

2
6 3
010100
6 2
010100

Sample Output

5
4

HINT

Source

hpu

看到巨巨的做法是，用数组记录0的位置，最后找第i个0和第i+k+1个0之间最大的长度，另外在最开始和结束再加一个0，便于计算开始和结束有k个0的情况。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int a[100010];
char s[100010];
int main()
{
	int t,n,k,cnt,i,j;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		scanf("%s",&s);
		cnt=0;
		a[cnt++]=0;
		for(i=0;i<n;i++)
		{
			if(s[i]=='0')
			{
				a[cnt++]=i+1;
			}
		}
		if(cnt-1<=k)
		{
			printf("%d\n",n);
			continue;
		}
	    a[cnt++]=n+1;
	    int num=-1;
	    for(i=k+1;i<cnt;i++)
	    {
	    	num=max(num,a[i]-a[i-k-1]-1);
		}
		printf("%d\n",num);
	}
	return 0;
}