Contest3109 - 2021级新生个人训练赛第20场_问题 H: 黑白奶牛

//
问题 H: 黑白奶牛

题目描述
有N只奶牛从左往右排成一行，编号是1至N。这N只奶牛当中，有一些奶牛是黑色的，其余的是白色的。
color[i]表示第i只奶牛的颜色，如果color[i]=0则表示第i头奶牛是黑色的，如果color[i]=1则表示第i头奶牛是白色的。
六一奶牛儿童节快到了，农场主Farmer John要从这N头奶牛当中，挑选尽可能多的奶牛去参加晚会。
Farmer John挑选奶牛的原则是：挑选编号是连续的一段奶牛，这一段奶牛的颜色必须全部是白色的。
Farmer John有一个魔法棒，每用一次魔法棒就可以把一头黑色的奶牛变成一头白色的奶牛，魔法棒最多只能使用K次。
在上述条件下，最多可以有多少头奶牛去参加晚会呢？
输入
第一行，两个整数，N和K。
第二行，N个整数，第i个整数就是color[i]，color[i]要么是0，要么是1。
输出
一个整数，表示最多有多少头奶牛可以去参加晚会。
样例输入 Copy
【样例1】
11 0
1 1 0 0 1 1 1 1 0 1 1
【样例2】
11 1
1 1 0 0 1 1 1 1 0 1 1
样例输出 Copy
【样例1】
4
【样例2】
7
提示
样例1解释：由于K=0，所以不能使用魔法棒，所以挑选编号是5至8的奶牛去参加晚会。
样例2解释：由于K=1，所以最多可以使用1 次魔法棒，使用魔法棒把第9 头奶牛变成白色奶牛，然后挑选编号是5至11 的奶牛去参加晚会。
对于50%的数据，1 <= N <= 1000，K = 0，即不能使用魔法棒。
对于100%的数据，1 <= N <= 100000，  1 <= K <=N。

//
题意：找到 在使用魔法棒后 连续白牛的最大长度
//
题解：
01  头尾标记 推进 动态区间 即可

// 黑白奶牛
#include<bits/stdc++.h>
using namespace std;

// 对于100%的数据，1 <= N <= 100000，  1 <= K <=N
const int N=100005;
int a[N];

int main()
{
    int n,k,i;
    int begin,end,count,ans;

    scanf("%d%d",&n,&k);

    for( i=0;i<n;i++ )
    {
        scanf("%d",&a[i]);
    }
    // 初始化
    begin=end=count=ans=0; 

    // 动态区间
    while( end<n )
    {
        // 尾标记推进：出现黑牛 计数
        if( a[ end++ ]==0 ) count++;

        // 头标记推进：将当前黑牛数量 控制在魔法棒数量范围内
        while( count>k )
        {
            // 找到黑牛 才减少数量
            if( a[ begin++ ]==0 ) count--;
        }
        // 当前最大值比较
        ans=max( ans,end-begin );
    }
    printf("%d\n",ans);

    return 0;
}
/**/

个人见解  酌情采纳.

标签：20,color,Contest3109,魔法,样例,级新生,晚会,挑选,奶牛
来源： https://blog.csdn.net/qq_63173957/article/details/122558378