【排序】奶牛的编号

题目描述

有N(1≤N≤1000)头奶牛，它们都被标上一个优先等级编号：1，2或3。用来表示它们喝水时的优先次序，编号为l的最优先，编号为2的其次，编号为3的最后。每天奶牛开始时排成一行，但总是很乱，需要你把它们重新排成编号为1的奶牛在最前面，编号为2的其次，编号为3的奶牛在最后。你能计算出最少需要多少的交换次序来完成这次重排吗？

 

输入

第1行：1个整数N；
第2至N+I行：第i+l行有一个整数表示开始队列中第i头奶牛的编号。

 

输出

1行，只一个整数，表示最少需要交换次数。

 

样例输入

9
2
2
1
3
3
3
2
3
1

 

样例输出

4

 

提示

 

样例说明：有一种交换方法。
2 2 2< 1 1
2< 1 1 1 1
1< 2 2 2 2
3 3< 2 2 2
3 3 3 3< 2
3 3 3 3 3
2 2< 3 3 3
3 3 3 3 3
1 1 1< 2< 3

 

我们可以定义一个cnt[]来记录1，2，3的个数，然后分块来讨论。首先在1的区域（即1~cnt[1]）遍历，如果a[i]==2，那么就去2的区域找1并替换，如果2的区域没有1，就去3的区域；如果a[i]==3，那么就去3的区域找1并替换，如果3的区域没有1，就去2的区域。接着就在2的区域（即cnt[1]+1~cnt[1]+cnt[2]），因为我们此时已经确保了1的区域只有1，那么2只能出现在2和3的区域，所以我们只需要看此时2的区域中有多少个3就行了。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
int n,a[1005],cnt[5];
int main()
{
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        cnt[a[i]]++;
    }
    int ans=0;
    for(int i=1;i<=cnt[1];i++)
    {
        if(a[i]==2)
        {
            for(int j=cnt[1]+1;j<=cnt[2]+cnt[1];j++)
            {
                if(a[j]==1)
                {
                    ans++;
                    swap(a[i],a[j]);
                    break;
                }
            }
            if(a[i]==2)
            {
                for(int j=cnt[1]+cnt[2]+1;j<=n;j++)
                {
                    if(a[j]==1)
                    {
                        ans++;
                        swap(a[i],a[j]);
                        break;
                    }
                }
            }
        }
        else if(a[i]==3)
        {
            for(int j=cnt[1]+cnt[2]+1;j<=n;j++)
            {
                if(a[j]==1)
                {
                    ans++;
                    swap(a[i],a[j]);
                    break;
                }
            }
            if(a[i]==3)
            {
                for(int j=cnt[1]+1;j<=cnt[2]+cnt[1];j++)
                {
                    if(a[j]==1)
                    {
                        ans++;
                        swap(a[i],a[j]);
                        break;
                    }
                }
            }
        }
    }
    for(int i=cnt[1]+1;i<=cnt[1]+cnt[2];i++)
    {
        if(a[i]!=2)
            ans++;
    }
    cout<<ans<<endl;
}