CodeForces 660C Hard Process

给定一个数组 a，含有 n 个元素。数组 a 中的每个元素要么是 0，要么是 1 。

让我们假定，数组 a 中，仅由数字 1 组成的连续元素所构成的子分段，其最长的长度为 f(a)。您可以将不超过 k 个 0 更改为 1，使得 f(a) 最大化。

输入

第一行包含了两个整数 n 和 k (1 ≤ n ≤ 3·10⁵, 0 ≤ k ≤ n) — 数组 a 中的元素数目，和参数 k 。

第二行包含了 n 个整数 a_i (0 ≤ a_i ≤ 1) — 数组 a 的元素。

输出

在第一行中，打印一个非零的整数 z — 在将不超过 k 个 0 更改为 1 之后，f(a) 的最大值。

在第二行中，打印 n 个整数 a_j — 在更改之后，数组 a 的各元素。

如果存在多个答案，请打印它们中的任何一种。

示例

输入

7 1
1 0 0 1 1 0 1

输出

4
1 0 0 1 1 1 1

输入

10 2
1 0 0 1 0 1 0 1 0 1

输出

5
1 0 0 1 1 1 1 1 0 1

一看到01数组，首先想到计数问题，然后想到前缀和，然后二分f(a)就可以了。二分的上界是n，下界是0，这道题没有什么难想的点吧，我感觉前缀和的方法应该比较简单好想一点，然后用二分小区间的写法，避免了很多麻烦，具体请看代码：

#include <stdio.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
const int maxn = 3*1e5+5;
int a[maxn];
int num[maxn];
int n, k, t;
int judge(int x) { //区间判断
    for(int i = 0; i <= n-x; i++) {
        if(num[i+x]-num[i]+k >= x) {
            t = i;
            return 1;
        }
    }
    return 0;
}
int main() {
    while(~scanf("%d %d", &n, &k)) {
        int sum = 0;    
        memset(num, 0, sizeof(num));
        for(int i = 1; i <= n; i++) {
            scanf("%d", &a[i]);
            sum += a[i];
            num[i] = sum;//求得前缀和
        }
        int left = 0, right = n, mid;
        while(left+4 < right) {  //二分小区间写法0_0,自创哦
            mid = (right + left)/2;
            if(judge(mid)) left = mid;
            else right = mid;
        }
        int ans = 0;
        for(int i = right; i >= left; i--) {//小区间枚举
            if(judge(i)) {
                for(int j = t+1; j <= t+i; j++) { //将这段区间全部赋值为1
                    ans++;
                    a[j] = 1;
                }
                break;
            }
        }
        printf("%d\n", ans); //输出结果
        for(int i = 1; i <= n; i++) {
            printf("%d", a[i]);
            if(i != n) printf(" ");
            else printf("\n");
        }
    }
    return 0;
}