【zzuli-oj】-1895-985的0-1串难题（思维，好）

1895: 985的0-1串难题

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 202  Solved: 53

SubmitStatusWeb
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

26 30101006 2010100

Sample Output

54

题解：在串的前后各加一个0，用一个数组op[i]记录0的个数和每个0 出现的位置。个数ant从0开始，第一个0记为op[0]=0，位置在0。后面遇到0，记录位置都向后移动一位。末尾添加的0位置是op[]=n+1;

比较ant-2与k的大小关系：1，小于就是n个。2，大于k，取k个0之间最长的串，更新maxx。

代码如下：

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
#define M 100000
#define inf 0x3f3f3f3f
char a[M+10];
int op[M+10]; //op[i]表示第i个 0 在第op[i]的位置
int main()
{
int u;
scanf("%d",&u);
while(u--)
{
int n,k,ant=0;
scanf("%d %d",&n,&k);
scanf("%s",a);
op[ant++]=0; //前加一个 0
for(int i=0;i<n;i++)
if(a[i]=='0')
op[ant++]=i+1; //前面加一个 0 ，后面 0 的位置统一右移一个
op[ant++]=n+1; //后加一个 0
if(k>=(ant-2))
printf("%d\n",n);
else
{
int maxx=0;
for(int i=k+1;i<ant;i++)
maxx=max(maxx,op[i]-op[i-k-1]-1); //求两个 0 之间的最长串
printf("%d\n",maxx);
}
}
return 0;
}