Codeforences Educational Round 11

【题目链接】点击打开链接

【题意】给了n,k，已经一个长度为n的01序列，问你在最多进行k此操作，(操作是把0变成1),能得到最大的连续1的长度。

【分析】维护0的前缀和，对每个以i点开头的字串，二分查找能到达的最大位置，维护最大的答案。

【AC代码】

#include <bits/stdc++.h>
using namespace std;
int a[300005],dp[300005];
bool vis[300005];
int main()
{
    int n,k;
    cin>>n>>k;
    memset(dp,0,sizeof(dp));
    memset(vis,false,sizeof(vis));
    for(int i=1; i<=n; i++)
    {
        scanf("%d",&a[i]);
        if(a[i]==0) dp[i]=dp[i-1]+1;
        else        dp[i]=dp[i-1];
    }
    int max_ans=0;
    for(int i=1; i<=n; i++)
    {
       int val=dp[i]+k;
       if(a[i]==0)val--;
       int pos = upper_bound(dp+1,dp+n+1,val)-(dp+1);
       if(pos==-1)
       {
           max_ans = max(max_ans,n-i);
       }
       else
       {
           max_ans = max(max_ans,pos-i+1);
       }
    }
    printf("%d\n",max_ans);
    for(int i=1; i<=n; i++)
    {
       int val=dp[i]+k;
       if(a[i]==0)val--;
       int pos = upper_bound(dp+1,dp+n+1,val)-(dp+1);
       if(pos==-1)
       {
           max_ans = max(max_ans,n-i);
           if(max_ans==n-i)
           {
               for(int j=i;j<=n;j++)vis[j]=true;
               break;
           }
       }
       else
       {
           max_ans = max(max_ans,pos-i+1);
           if(max_ans==pos-i+1)
           {
               for(int j=i;j<=pos;j++)vis[j]=true;
               break;
           }
       }
    }
    for(int i=1; i<=n; i++)
    {
        if(vis[i])printf("1 ");
        else printf("%d ",a[i]);
    }
    return 0;
}