题意:先给你一个n,k,n表示下面会有n个数字,这些数字只有0和1,k表示你最多可以将k个0变为1。问变完后连续的最长的全是1的串的长度是多少,并且输出最后得串;
思路:我们可以做一个有几个0的前缀和sum,然后枚举左端点二分查找含有k个0的序列的右端点,找到一个最长的序列就是答案。
#include<bits/stdc++.h>
using namespace std;
const int MAXN = 3e5 + 5;
int n,k,a[MAXN],sum[MAXN];
int main()
{
while(~scanf("%d%d",&n,&k))
{
memset(sum,0,sizeof(sum));
for(int i = 1; i <= n; i++) scanf("%d",&a[i]);
for(int i = 1; i <= n; i++)
{
sum[i] = sum[i - 1];
if(a[i] == 0) sum[i]++;
}
int MAX = -1,L = -1,R = -1;//最长的序列长度,左右端点
for(int i = 1; i <= n; i++)
{
//int pos = upper_bound(sum + i,sum + n,sum[i - 1] + k) - sum;
int pos = -1;
int l = i,r = n,m = l + (r - l) / 2,key = sum[i - 1] + k;
while(l < r)//二分找上界
{
if(sum[m] <= key) l = m + 1;
else r = m;
m = l + (r - l) / 2;
}
if(sum[m] <= key) pos = m + 1;//上界不存在
else pos = m;//存在
if(pos - i> MAX)
{
MAX = pos - i;L = i;R = pos;
}
}
printf("%d\n",MAX);
for(int i = L; i < R; i++) a[i] = 1;
for(int i = 1; i <= n; i++) printf("%d ",a[i]);
printf("\n");
}
return 0;
}