给定一个数组 a,含有 n 个元素。数组 a 中的每个元素要么是 0,要么是 1 。
让我们假定,数组 a 中,仅由数字 1 组成的连续元素所构成的子分段,其最长的长度为 f(a)。您可以将不超过 k 个 0 更改为 1,使得 f(a) 最大化。
第一行包含了两个整数 n 和 k (1 ≤ n ≤ 3·105, 0 ≤ k ≤ n) — 数组 a 中的元素数目,和参数 k 。
第二行包含了 n 个整数 ai (0 ≤ ai ≤ 1) — 数组 a 的元素。
在第一行中,打印一个非零的整数 z — 在将不超过 k 个 0 更改为 1 之后,f(a) 的最大值。
在第二行中,打印 n 个整数 aj — 在更改之后,数组 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;
}