“玲珑杯”线上赛 Round #15 河南专场 咸鱼魔法记

DESCRIPTION

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

INPUT

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

OUTPUT

　　输出答案。

 

SAMPLE INPUT

　　10 2

　　1 0 0 1 0 1 0 1 0 1

SAMPLE OUTPUT

　　5

 

 

前缀和+二分处理，先构造一个前缀和，再完后二分处理。

#include <iostream>
#include <stdio.h>
#include <string.h>
using namespace std;
const int MAX = 300010;
int ans[MAX], n, k, num;
int check(int pos){
    int l = pos, r = n;
    while(l <= r){
        int m = (l+r)>>1;
        if(m-pos+1 <= k+ans[m]-ans[pos-1])l=m+1;
        else r = m-1;
    }
    return l-1;
}
int main(){
    scanf("%d%d",&n,&k);
    for(int i = 1; i <= n; i ++)
        scanf("%d",&num), ans[i] = ans[i-1]+num;
    int Max = 0;
    for(int i = 1; i <= n; i ++){
        int x = check(i);
    //    printf("%d   %d\n",i,x);
        if(Max < (x-i+1)){
            Max = x-i+1;
        }
    }
    printf("%d\n",Max);
    return 0;
}