最长全1串

题目：给你一个01串，定义答案=该串中最长的连续为1的长度，现在你有至多k次机会，每次机会可以将串中的某个0改成1，现在问最大可能答案。
输入：

输入第一行的两个整数N,k，表示字符串长度和机会次数
第二行输入N个整数，表示该字符串的元素
1
2
输出

输出一行表示答案
1
样例输入

10 2
1 0 0 1 0 1 0 1 0 1
1
2
样例输出

5
思路：

           先记录下每个0的位置，然后以k长度从左往右滑动；

           每滑动一次就在两头分别设置一个指针，分别往左和往右，遇到第一个0便终止；

           从中取最大值。

#include <stdio.h>
#include <iostream>
#include<cstring>
#include <vector>
#include <assert.h>
#include <queue>
using namespace std;
#define MAX 300005
int arr[MAX];
int num[MAX];
int process(int n, int k, int numZ, int cnt){
    if(numZ <= k) return n;
    int top = 0;
    int ans = 0;

    for(; top<cnt; top++){
        int i = arr[top];

        int sum = top + k < cnt ? arr[top+k-1] - i + 1: n - i;
        for(int j = i - 1; j >= 0; j--){
            if(i <= 0) break;
            if(num[j]) sum++;
            else break;
        }
        //cout<<"#"<<i<<" l "<<sum<<endl;
        for(int j = top + k < cnt ? arr[top+k-1] + 1: n - i; j < n; j++){
            if(j == n-i ) break;
            if(num[j]) sum++;
            else break;
        }
        //cout<<"#"<<i<<" r "<<sum<<endl;
        ans = max(ans, sum);

    }

    return ans;
}
int main(){
    int n, k;

    cin>>n>>k;
    for(int i = 0; i < n; i++) cin>>num[i];
    int top = 0;
    int numZ = 0;
    for(int i = 0; i < n; i++){
        if(!num[i]){
            numZ ++;
            arr[top++]=i;
        }
    }
    cout<<process(n, k, numZ, top)<<endl;
    return 0;
}
/*
10 2
1 0 0 1 0 1 0 1 0 1
*/