n 只奶牛坐在一排,每个奶牛拥有 ai 个苹果,现在你要在它们之间转移苹果,使得最后所有奶牛拥有的苹果数都相同,每一次,你只能从一只奶牛身上拿走恰好两个苹果到另一个奶牛上,问最少需要移动多少次可以平分苹果,如果方案不存在输出 -1。
输入描述:
每个输入包含一个测试用例。每个测试用例的第一行包含一个整数 n(1 <= n <= 100),接下来的一行包含 n 个整数 ai(1 <= ai <= 100)。
输出描述:
输出一行表示最少需要移动多少次可以平分苹果,如果方案不存在则输出 -1。
输入例子:
4
7 15 9 5
输出例子:
3
#include <iostream>
#include <string>
#include <cstring>
#include <set>
#include <cmath>
using namespace std;
int const maxn = 110;
int n;
int data[maxn];
int ans = 2<<10;
int p=0;
int sum = 0;
int aver;
int flag = 0;
void dfs(int i) {
if (i==n) {
if (p<ans) ans= p;
flag = 1;
} else if(flag==0)
{
{
if(data[i]>=aver) dfs(i+1);
else {
for (int j=0;j<n;j++) {
if(data[j]>aver&&data[i]<aver&&data[j]-aver>=aver-data[i]) {
int tmp = aver-data[i];
data[j] -= aver-data[i];
p+=(tmp)/2;
data[i] = aver;
dfs(i+1);
data[j] += tmp;
p-=tmp/2;
data[i]= aver-tmp;
} else if(data[j]>aver&&data[i]<aver&&data[j]-aver<aver-data[i]) {
int tmp = data[j]-aver;
data[j] = aver;
p+=(tmp)/2;
data[i] +=tmp;
dfs(i);
data[j] =aver+tmp;
p-=tmp/2;
data[i] -= tmp;
}
}
}
}
}
}
bool judge() {
if(sum%n!=0) return 0;
if(sum%n==0) {
for (int i=0;i<n;i++) {
if(abs(data[i]-aver)%2!=0) return 0;
}
}
return 1;
}
int main() {
cin>>n;
for (int i=0;i<n;i++) {
cin>>data[i];
sum+=data[i];}
aver = sum/n;
if (judge()) {
dfs(0);
cout<<ans<<endl;
} else {
cout<<-1<<endl;
}
}