18709 魔法

时间限制:1000MS  代码长度限制:10KB
提交次数:0 通过次数:0

题型: 编程题 语言: G++;GCC

Description

农夫约翰的奶牛场有很多奶牛，奶牛有黑白两种颜色。现在奶牛们排成整齐的一列去参加镇上的游行活动。
约翰希望白色奶牛都排在前面，黑色的奶牛都排在后面。但现在队列中奶牛的颜色是混乱的，并且奶牛们都不愿意改变位置。
幸运的是，约翰有一根魔法棒，每挥舞一次魔法棒就可以改变一头奶牛的颜色。
请问，约翰至少要挥舞多少次魔法棒，才能将队列改成他希望的状态。注意，可以将所有的奶牛都变成白色，或者都变成黑色。

 

输入格式

第一行一个正整数n，表示奶牛的头数。(1<=n<=200000)。
第二行n个正整数，均为1或2，1表示白色奶牛，2表示黑色奶牛。

 

输出格式

一个正整数，表示挥舞魔法棒的最少次数。

 

输入样例

7
2 2 1 1 1 2 1

 

输出样例

3

 

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;
const int N = 1e6;
int a[N],black_front[N];

int main()
{   ios::sync_with_stdio(0),cin.tie(0);
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        if(a[i]==2)black_front[i]=black_front[i-1]+1;
        else black_front[i]=black_front[i-1];
    }
    int ans=200001;
    for (int i = 1; i <= n; i ++ )
    ans=min(ans,black_front[i]+n-i-(black_front[n]-black_front[i]));
    //n-i为后面牛的数量，减去后面黑牛数量 就是i后白牛数量，然后迭代贪心
    
    cout<<ans;

   return 0;
}

**注意n的值超过1e4，暴力解n2会超时，可用前缀和思想优化