[每日一题]83：分苹果（贪心）

题目描述

n 只奶牛坐在一排，每个奶牛拥有 ai 个苹果，现在你要在它们之间转移苹果，使得最后所有奶牛拥有的苹果数都相同，每一次，你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上，问最少需要移动多少次可以平分苹果，如果方案不存在输出 -1。

输入描述:

每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n（1 <= n <= 100），接下来的一行包含 n 个整数 ai（1 <= ai <= 100）。

输出描述:

输出一行表示最少需要移动多少次可以平分苹果，如果方案不存在则输出 -1。

示例1

输入

4
7 15 9 5
输出

3

题解思路

首先说明两种不能平分的情况

• 第一：苹果总数不能整除牛的数量
• 第二：牛拥有的苹果数与苹果的平均数相差奇数

如果不属于以上两种情况，可以将苹果数多于平均数的牛的苹果分出去，计算分的次数就可以了。

代码示例：

#include <iostream>
#include <vector>
using namespace std;

int main(){
    int n;
    while(cin >> n){
        vector<int> v(n);
        int sum = 0;
        for(int i = 0; i < n; i++){
            cin >> v[i];
            sum += v[i];
        }
        
        if(sum % n){
            cout << "-1" << endl;
            break;
        }
        
        int tmp = sum / n , flag = 0;
        for(auto i : v){
            if((i - tmp) % 2){
                cout << "-1" << endl;
                flag = 1;
                break;
            }
        }
        
        if(!flag){
            int count = 0;
            for(auto i : v){
                if(i > tmp){
                    count += (i - tmp) / 2;
                }
            }
            cout << count << endl;
        }
    }
    return 0;
}