题意:有n个人坐在圆桌上,每个人有初始金币m[i],金币总数能整除n,每个人可以给相邻的人金币,也可以得到相邻的人给的金币,问最少转手多少金币可以使所有人的金币数量相等。
题解:最后每个人手上应该有sum / n个金币,设每个人i得到和失去的金币分别是x[i]和x[i + 1],那么可以得到所有转移金币是x1 、x1 - (m1 - sum / n)、x1 - (m1 + m2 - 2 * sum / n) ...,如果所有的x绝对值最小,那么转手金币就是最小的,所以问题转化为求极值,有个结论是所有数到中位数点的距离和是最小的。所以当x1是所有被减数的中位数时最小,求出距离和就是结果。
#include <stdio.h>
#include <cmath>
#include <algorithm>
#define ll long long
using namespace std;
const int N = 1000005;
ll m[N], c[N];
int main() {
int n;
while (scanf("%d", &n) == 1) {
ll sum = 0;
for (int i = 1; i <= n; i++) {
scanf("%lld", &m[i]);
sum += m[i];
}
ll temp = sum / n;
c[0] = 0;
for (int i = 1; i < n; i++)
c[i] = c[i - 1] + m[i] - temp;
sort(c, c + n);
ll x1 = c[n / 2], res = 0;
for (int i = 0; i < n; i++)
res += abs(c[i] - x1);
printf("%lld\n", res);
}
return 0;
}