【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
;
for(int i=0;i<n;i++)
cin>>arr[i];
cout<<solve(arr,n)<<endl;
delete []arr;
return 0;
}