2-5 均等笔

2-5 均等笔 (100分)

n个人围成一圈，每人有ai支笔。每人可以向左右相邻的人传递笔，每人每次传递一支笔消耗的能量为1。求使所有人获得均等数量的笔的最小能量。

输入格式:
第一行一个整数n ，表示人的个数（30%的数据，n<=1000；100%的数据，n<=1e6）。
接下来n行，每行一个整数 ai。

输出格式:
输出一个整数，表示使所有人获得均等笔的最小能量。（答案保证可以用64位有符号整数存储）

输入样例:

4
1
2
5
4

输出样例:

4

代码：

#include<stdio.h>
#include<stdlib.h>

long long n, ans, sum, ave, mid, a[1000005], x[1000005];

int cmp(const void* a, const void* b) { return *(int*)a - *(int*)b; }

int main() 
{
	scanf("%lld", &n);
	for (int i = 0; i < n; i++) {
		scanf("%lld", &a[i]);
		sum += a[i];
	}
	ave = sum / n;
	x[0] = a[0] - ave;
	for (int i = 1; i < n; i++)
		x[i] = a[i] + x[i - 1] - ave;
	qsort(x,n,sizeof(x[0]),cmp);
	for (int i = 0; i < n / 2; i++)
		ans += x[n - 1 - i] - x[i];
	printf("%lld", ans);

	return 0;
}