“玲珑杯”线上赛 Round #15 咸鱼魔法记（二分搜索）

DESCRIPTION

给你一个01串，我们定义这个串的咸鱼值，是最长的全1串。现在你最多可以使用K次咸鱼魔法，每次魔法，你可以使得一个位置翻转（0变成1,1变成0）。问你这个串的咸鱼值最多是多少。

INPUT

第一行两个整数N,K。表示串的长度和可以施展咸鱼魔法的次数。(N,K<=300000)第二行N个01整数。

OUTPUT

输出答案。

SAMPLE INPUT

10 21 0 0 1 0 1 0 1 0 1

SAMPLE OUTPUT

5

思路：可以确定咸鱼值区间为[0,N]，所以二分搜索即可；

ps:搜索判断用了尺取的思想，即本题也可以通过尺取法不断更新来写；

代码：

#include<bits/stdc++.h>
using namespace std;

int a[1000007];
int n,k;

bool C(int x)
{
    int sum = 0;
    for(int i=0; i<x; i++)
        sum += a[i];
    if(sum + k >= x)
        return true;
    for(int i=x; i<n; i++)//尺取前进保证是x个连续数 
    {
        sum += a[i] - a[i-x];
        if(sum + k >= x)
            return true;
    }
    return false;
}

int main()
{
    scanf("%d%d", &n, &k); 
    for(int i=0; i<n; i++)
        scanf("%d", &a[i]);
    int lb = 0, ub = n;
    while(lb < ub)
    {
        int mid = (lb + ub + 1) / 2;
        if(C(mid))
            lb = mid;
        else
            ub = mid - 1;
    }
    printf("%d\n",lb);
    return 0;
}