- 题目
分苹果
时间限制: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;
}