ZZuli oj 1895: 985的0-1串难题(二分+打表）

1895: 985的0-1串难题

Time Limit: 1 Sec   Memory Limit: 128 MB
Submit: 259   Solved: 74

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

思路：先打表，打出前i个数中0的个数，然后二分判断每个mid长度的串中0的个数是否小于k;

代码：

#include<stdio.h>
#include<string.h>
char s[100010];
int biao[100010];//biao[i]表示前i个数（包括i)中0的个数； 
int n,k,t;
void dabiao()//打表打出0的个数； 
{
	memset(biao,0,sizeof(biao));
	if(s[0]=='0')
	   biao[0]=1;
	for(int i=1;i<n;i++)
	{
		//biao[i]=biao[i-1]+(s[i]=='0'?1:0);耗时间； 
		if(s[i]=='0')
		   biao[i]=biao[i-1]+1;
		else
		  biao[i]=biao[i-1];
	}
}
bool judge(int mid)
{
	int size=n-mid;
	for(int i=0;i<=size;i++)
	{
		int temp=0;
		if(s[i]=='0')
		  temp++;
		if(biao[i+mid-1]-biao[i]+temp<=k)//表示长度为mid的串中0的个数； 
		   return true;
	}
	return false;
}
int main()
{
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&n,&k);
		scanf("%s",s);
		dabiao();
		int l=1,r=n,ans;
		while(l<=r)
		{
			int mid=(l+r)/2;
			if(judge(mid))
			{
				ans=mid;
				l=mid+1;
			}
			else
			{
				r=mid-1;
			}
		}
		printf("%d\n",ans);
	}
	return 0;
}