Spreading the Wealth

题目:

n个人围成一圈,传递现有金额,是所有人平均.

思路:

对于第一个人, 有A1-x1+x2=average;

对于第二个人,有A2-x2+x3=average;

其中,x1, x2, x3,为相应的人分配出去的量.

那么可以推出:

x2 = average-A1+x1=x1-c1,c1=A1-average;

x3=average-A2+x2=2average-A1-A2+x1=x1-c2, c2=c1-average+A2;

课的出地推公式.

最后要求x1到c1,c2..cn各点距离最小,即x1是中位数,通过x1与各分配量的关系(c)求得值.

代码:

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int m[1000005], c[1000005];
int main() {
	int n;
	while(scanf("%d", &n) != EOF) {
		long long sum = 0;
		for(int i=1; i<=n; i++) {
			scanf("%d", &m[i]);
			sum += m[i];
		}
		long long aver = sum/n;
		c[0] = 0;
		for(int i=1; i<=n; i++) {
			c[i] = c[i-1] + m[i] - aver; 
		}
		sort(c, c+n);
		long long x = c[n/2];
		long long minx = 0;
		for(int i=0; i<n; i++) {
			if(x > c[i]) minx+=(x-c[i]);
			else minx += (c[i]-x);
		}
		printf("%lld\n", minx);
	}
	return 0;
}