网易秋招笔试题 - 分苹果

• 题目
  分苹果
  时间限制：1秒 空间限制：32768K 热度指数：44995
  算法知识视频讲解
  校招时部分企业笔试将禁止编程题跳出页面，为提前适应，练习时请使用在线自测，而非本地IDE。
  题目描述
  n 只奶牛坐在一排，每个奶牛拥有 ai 个苹果，现在你要在它们之间转移苹果，使得最后所有奶牛拥有的苹果数都相同，每一次，你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上，问最少需要移动多少次可以平分苹果，如果方案不存在输出 -1。
  输入描述:
  每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n（1 <= n <= 100），接下来的一行包含 n 个整数 ai（1 <= ai <= 100）。
  输出描述:
  输出一行表示最少需要移动多少次可以平分苹果，如果方案不存在则输出 -1。
  示例1
  输入
  复制
  4
  7 15 9 5
  输出
  复制
  3
• 解法
  代码

#include<iostream>
#include<algorithm>

using namespace std;


int get_sum(int* num,int count);

int main()
{
    int n;
    cin>>n;
    
    int* num=new int[n];
    
    int i,result,sum,ave,id1,id2;
    i=0;
    while(i<n)
    {
        cin>>num[i];
        i++;
    }
 
    sum=get_sum(num,n);
    
    if(sum%n!=0)
    {
        cout<<-1<<endl;
        return 0;
    }
    ave=sum/n;
    for(int j=0;j<n;j++)
    {
        if(num[j]>=ave&&(num[j]-ave)%2!=0)
        {
            cout<<-1<<endl;
            return 0;
        }
        if(num[j]<=ave&&(ave-num[j])%2!=0)
        {
            cout<<-1<<endl;
            return 0;
        }
    }
      
    id1=0;
    id2=0;
    result=0;
    while(id1<n||id2<n)
    {
        while(num[id1]>=ave&&id1<n)
            id1++;
        while(num[id2]<=ave&&id2<n)
            id2++;
        if(id1<n&&id2<n)
        {
            num[id1]+=2;
            num[id2]-=2;
            result++;
        }
    }
    
    cout<<result<<endl;

    return 0;
}

int get_sum(int* num,int count)
{
    int result;
    result=0;
    for(int i=0;i<count;i++)
        result+=num[i];
    return result;
}