1.求最值问题考虑二分
2.一维数组多想想前缀和
3.一维数组区间问题可以考虑双指针(尺取):枚举右端点,求左端点
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
int a[1000001],sum[1000001];
int n,k;
bool check(int x)
{
for(int i=1;i+x-1<=n;i++)
{
if(sum[i+x-1]-sum[i-1]<=k)
{
return 1;
}
}
return 0;
}
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%d",&a[i]);
sum[i]=sum[i-1]+(a[i]==0);
}
int l=0,r=n;
while(l<r)
{
int mid=(l+r+1)/2;
if(check(mid)) l=mid;
else r=mid-1;
}
printf("%d\n",l);
for(int i=1;i+l-1<=n;i++)
{
if(sum[i+l-1]-sum[i-1]<=k)
{
for(int j=i;j<=i+l-1;j++) a[j]=1;
break;
}
}
for(int i=1;i<=n;i++)
{
if(i==1) printf("%d",a[i]);
else printf(" %d",a[i]);
}
printf("\n");
}