网易-分苹果

​​https://www.nowcoder.com/practice/a174820de48147d489f64103af152709?tpId=85&tqId=29834&tPage=1&rp=1&ru=/ta/2017test&qru=/ta/2017test/question-ranking​​

题目描述

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> a(n, 0);
    int sum = 0;
    for (int i = 0; i < n; i++){
      cin >> a[i];
      sum += a[i];
    }
    if (sum % n != 0){
      cout << "-1" << endl;
    }
    else {
      bool isOk = true;
      int avg = sum / n;
      int cnt = 0;
      for (int i = 0; i < n; i++){
        if ((a[i] - avg) % 2 != 0){
          cout << "-1" << endl;
          isOk = false;
          break;
        }
      }
      if (isOk == true){
        for (int i = 0; i < n; i++){
          if (a[i] - avg > 0){
            cnt += (a[i] - avg) / 2;
          }
        }
        cout << cnt << endl;
      }
    }
  }
  return 0;
}