【zzuliOJ】1895 - 985的0-1串难题（思维）

点击打开题目

1895: 985的0-1串难题

Time Limit: 1 Sec  Memory Limit: 128 MB
Submit: 114  Solved: 20

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

2

6 3

010100

6 2

010100

Sample Output

54

HINT

Source

hpu

这道题比赛的时候没做出来确实太不应该了。

思路不是很难想，在串的最前和最后都加上一个0，然后记录下每个0个位置，然后依次枚举两个0中间的串，使之最长即可。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define CLR(a,b) memset(a,b,sizeof(a))
char a[100000+11];
int pos[100000+11];
int main()
{
int n,k;
int ant;
int u;
scanf ("%d",&u);
while (u--)
{
scanf ("%d %d",&n,&k);
scanf ("%s",a);
ant = 0;
pos[ant++] = 0;
for (int i = 0 ; i < n ; i++)
if (a[i] == '0')
pos[ant++] = i+1;
pos[ant++] = n+1; //前面加个0后面加个零
if (k >= ant - 2) //k比0个个数多，则全变为1
{
printf ("%d\n",n);
continue;
}
int ans = 0;
for (int i = k + 1 ; i < ant ; i++)
ans = max (ans , pos[i] - pos[i-k-1] - 1); //求两个0中间最长的串
printf ("%d\n",ans);
}
return 0;
}