牛客网-分苹果(网易)
题目描述
n 只奶牛坐在一排,每个奶牛拥有 ai 个苹果,现在你要在它们之间转移苹果,使得最后所有奶牛拥有的苹果数都相同,每一次,你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上,问最少需要移动多少次可以平分苹果,如果方案不存在输出 -1。
输入描述:
每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n(1 <= n <= 100),接下来的一行包含 n 个整数 ai
(1 <= ai <= 100)。
输出描述:
输出一行表示最少需要移动多少次可以平分苹果,如果方案不存在则输出 -1。
示例1
输入
4 7 15 9 5
输出
3
解题思路:
(1)首先计算所有奶牛拥有的苹果之和,除以奶牛个数,若无余数,说明是可以平分的,若有余数,返回-1;
(2)因为每一次只能从奶牛身上拿走两个苹果,所以需要计算每个奶牛拥有的苹果数减去平均数,看这个差是否为2的倍数,若不是,返回-1;若是,统计所有差大于0(或者小于0)的差之和,然后除以2,则为最小需要移动的次数
7 15 9 5 average=9;
-2 6 0 -4 正数之和等于负数之和的绝对值
找到所有为正数的值的和,为6, 6/2=3
代码如下:
#include<iostream> #include<vector> using namespace std; int Minimal(vector<int> vec,int n){ int sum=0; for(int i=0;i<n;++i) sum+=vec[i]; if(sum%n!=0) //判断是否能够平均 return -1; int average=sum/n; int num=0; for(int i=0;i<n;++i){ if((vec[i]-average)%2!=0){ //因为每次需要移动两个苹果,所以需要差为2的倍数 return -1; break; } else{ if((vec[i]-average)>0){//统计所有差为正数的差的和(或者统计所有差为负数的差的和) num+=vec[i]-average; } } } return num/2; //每次移动两个,计算移动的次数 } int main(){ int n; vector<int> vec(100); cin>>n; for(int i=0;i<n;++i) cin>>vec[i]; int result=Minimal(vec,n); cout<<result<<endl; system("pause"); return 0; }