【codeforces 779A】Pupils Redistribution

【题目链接】:http://codeforces.com/contest/779/problem/A

【题意】



让你把两个组的5个人的数目都变成一样的.

支持交换操作;

问你最少需要交换几次.

【题解】

/*
    哪个少了,就往对面的多的换一下;
    模拟一下写个贪心就好
    A:num[1..5]
    B:num[1..5]
    rep1(i,1,5)
    {
        if (bnum[i]>anum[i])
            {
                rep1(j,i,5)
                    swap(anum[j],bnum[j]);
            }
        if (bnum[i]<anum[i])
        {
            if ((bnum[i]+anum[i])&1)
                return puts("-1"),0;
            while (bnum[i]<anum[i])
            {
                int j = -1;
                rep1(k,i+1,5)
                    if (bnum[k]>anum[k] && (anum[k]+bnum[k])%2==0)
                        {
                            j = k;
                            break;
                        }
                if (j==-1)
                    return puts("-1"),0;
                bnum[j]--,anum[j]++;
                bnum[i]++,anum[i]--;
            }
        }
    }
*/

【完整代码】

#include <bits/stdc++.h>
using namespace std;
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
#define LL long long
#define rep1(i,a,b) for (int i = a;i <= b;i++)
#define rep2(i,a,b) for (int i = a;i >= b;i--)
#define mp make_pair
#define pb push_back
#define fi first
#define se second
#define rei(x) scanf("%d",&x)
#define rel(x) scanf("%lld",&x)
typedef pair<int, int> pii;
typedef pair<LL, LL> pll;
const int dx[9] = { 0,1,-1,0,0,-1,-1,1,1 };
const int dy[9] = { 0,0,0,-1,1,-1,1,-1,1 };
const double pi = acos(-1.0);
const int N = 110;
int anum[10],bnum[10];
int n,ans = 0;
int main()
{
    //freopen("F:\\rush.txt", "r", stdin);
    rei(n);
    rep1(i, 1, n)
    {
        int x;
        rei(x);
        anum[x]++;
    }
    rep1(i, 1, n)
    {
        int x;
        rei(x);
        bnum[x]++;
    }
    rep1(i, 1, 5)
    {
        if (bnum[i]>anum[i])
        {
            rep1(j, i, 5)
                swap(anum[j], bnum[j]);
        }
        if (bnum[i]<anum[i])
        {
            if ((bnum[i] + anum[i]) & 1)
                return puts("-1"), 0;
            while (bnum[i]<anum[i])
            {
                int j = -1;
                rep1(k, i + 1, 5)
                    if (bnum[k]>anum[k] && (anum[k] + bnum[k]) % 2 == 0)
                    {
                        j = k;
                        break;
                    }
                if (j == -1)
                    return puts("-1"), 0;
                bnum[j]--, anum[j]++;
                bnum[i]++, anum[i]--;
                ans++;
            }
        }
    }
    printf("%d\n", ans);
    return 0;
}