66 lines
1.9 KiB
C++
66 lines
1.9 KiB
C++
|
#include <bits/stdc++.h>
|
|||
|
|
using namespace std;
|
|||
|
|
const int N = 10010;
|
|||
|
|
int a[N];
|
|||
|
|
int sum, avg, total;
|
|||
|
|
int main() {
|
|||
|
|
int n;
|
|||
|
|
cin >> n;
|
|||
|
|
|
|||
|
|
for (int i = 1; i <= n; i++) {
|
|||
|
|
cin >> a[i];
|
|||
|
|
sum += a[i];
|
|||
|
|
}
|
|||
|
|
if (sum % n) {
|
|||
|
|
cout << -1 << endl;
|
|||
|
|
exit(0);
|
|||
|
|
}
|
|||
|
|
avg = sum / n; // 平均数
|
|||
|
|
|
|||
|
|
/*
|
|||
|
|
算法思路
|
|||
|
|
1、求出平均数avg
|
|||
|
|
2、枚举每个数字:
|
|||
|
|
(1) 如果当前数字比avg大，不断向后面找出比avg小的数，将自己多出的部分传递给这个数=min(超过avg的量,少于avg的量)
|
|||
|
|
如果一下子没用了，就继续向后做同样的操作
|
|||
|
|
(2) 如果当前数字比avg小，不断向后面找出比avg大的数，将自己少出的部分传递给这个数=min(超过avg的量,少于avg的量)
|
|||
|
|
如果一下子没用了，就继续向后做同样的操作
|
|||
|
|
|
|||
|
|
举栗子思考：
|
|||
|
|
3
|
|||
|
|
10 1 1
|
|||
|
|
|
|||
|
|
|
|||
|
|
3
|
|||
|
|
5 1 6
|
|||
|
|
|
|||
|
|
|
|||
|
|
3
|
|||
|
|
1 5 6
|
|||
|
|
*/
|
|||
|
|
for (int i = 1; i <= n; i++) {
|
|||
|
|
int d = a[i] - avg; // 差值，注意：这个差值可能是正的，也可能是负的
|
|||
|
|
total += abs(d);
|
|||
|
|
|
|||
|
|
for (int j = i + 1; j <= n; j++) { // 枚举i的所有后位的小朋友,如果当前i位上的小朋友需要糖果或者多出来糖果时，应该把
|
|||
|
|
// 多出来的糖果分给哪些小朋友，或者，少出来的糖果需要从哪些小朋友那里要过来
|
|||
|
|
// 向后找比avg小的
|
|||
|
|
if (d > 0 && a[j] < avg) {
|
|||
|
|
int x = min(d, avg - a[j]);
|
|||
|
|
a[i] -= x;
|
|||
|
|
a[j] += x;
|
|||
|
|
d -= x;
|
|||
|
|
}
|
|||
|
|
// 向后找比avg大的
|
|||
|
|
if (d < 0 && a[j] > avg) {
|
|||
|
|
int x = min(abs(d), a[j] - avg);
|
|||
|
|
a[i] += x;
|
|||
|
|
a[j] -= x;
|
|||
|
|
d += x;
|
|||
|
|
}
|
|||
|
|
if (d == 0) break;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
cout << total << endl;
|
|||
|
|
return 0;
|
|||
|
|
}
|