【newCoder】2017春招真题 [编程题] 分苹果

题目描述

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

示例1

输入
4
7 15 9 5
输出
3

代码如下

#include <iostream>
using namespace std;
int solve(int *arr,int n)
{
    int tot = 0;
    for(int i=0;i<n;i++)
        tot+=arr[i];
    int avg;
    if(tot%n==0)
        avg=tot/n;
    else
        return -1;//说明不存在

    int result=0;
    for(int i=0;i<n;i++)
    {
        if((arr[i]-avg)%2!=0)
            return -1;
        if(arr[i]>avg)
            result+=(arr[i]-avg)/2;
    }
    return result;
}
int main()
{
    int n;
    cin>>n;
    int *arr = new int[n];
    for(int i=0;i<n;i++)
        cin>>arr[i];
    cout<<solve(arr,n)<<endl;
    delete []arr;
    return 0;
}