题目描述
n 只奶牛坐在一排,每个奶牛拥有 a
i 个苹果,现在你要在它们之间转移苹果,使得最后所有奶牛拥有的苹果数都相同,每一次,你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上,问最少需要移动多少次可以平分苹果,如果方案不存在输出 -1。
输入描述:
每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n(1 <= n <= 100),接下来的一行包含 n 个整数 ai(1 <= ai <= 100)。
输出描述:
输出一行表示最少需要移动多少次可以平分苹果,如果方案不存在则输出 -1。
思路: 判断-1的情况: 给2的时候总是多1或者少1 平均数与数字之间的差值总是不为2的倍数
可以选择用快排从小到大
#include <iostream>
#include<cstdlib>
using namespace std;
int num[110];
int cmp(const void *a, const void *b){
return *(int *)a > *(int *)b; //函数返回值 大于0 则qsort排序认为前者“大于”后者 本题中按值从小到大排序
}
int main()
{
int n;
cin>>n;
int sum = 0;
for(int i=0; i<n; i++)
{
cin>>num[i];
sum += num[i];
}
if(sum%n)
{
cout<<-1;
return 0;
}
qsort(num, n, sizeof(int), cmp);
/*
for(int i=0; i<n; i++)
{
cout<<num[i]<<' ';
}
*/
int step = 0;
int avg = sum/n;
for(int i=n-1,j=0;i>=0&&j<=i;)
{
if(num[j] == avg && num[i] == avg)
break;
if((avg - num[j])%2 || (avg - num[i])%2)
{
cout<<-1;
return 0;
}
while(num[j]<avg && num[i]>avg)
{
num[j] += 2;
num[i] -= 2;
step++;
}
if(num[j] == avg)
j++;
if(num[i] == avg)
i--;
}
cout<<step<<endl;
//system("pause");
return 0;
}