BZOJ 1045: [HAOI2008] 糖果传递

/*
* 题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1045
* 有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传
* 递一个糖果代价为1，求使所有人获得均等糖果的最小代价。
*
* 分析：
* 假设a1分给an的糖果数为k，则可以得到以下的信息：
*                a1              a2                       　a3 　　　　　　　　   an-1 　　　　　　　　　　　　 an
* 当前数目：a1-k           a2 　　　　　　　　 a3 　　　　　　　　   an-1 　　　　　　　　　　　　 an+k
* 所需代价：|a1-k-ave| |a1+a2-k-2*ave| |a1+a2+a3-k-3*ave| |a1+..+a(n-1)-k-(n-1)*ave|   |k|
* 以sum[i]表示从a1加到ai减掉i*ave的和值，这以上可以化简为
* 总代价 = |s1-k|+|s2-k|+...+|s(n-1)-k|+|k|
* 不难看出：当k为s1...s(n-1)中的中位数的时候，所需的代价最小
*
* */

#include <cstdio>

#include <cstring>

#include <iostream>

#include <algorithm>



using namespace std;



const int X = 1000005;



typedef long long ll;



ll sum[X],a[X];

ll n;



ll Abs(ll x){

	return max(x,-x);

}



int main(){

	//freopen("sum.in","r",stdin);

	while(cin>>n){

		ll x;

		ll tot = 0;

		for(int i=1;i<=n;i++){

			scanf("%lld",&a[i]);

			tot += a[i];

		}

		

		ll ave = tot/n;

		for(int i=1;i<n;i++)

			sum[i] = a[i]+sum[i-1]-ave;

		sort(sum+1,sum+n);

		

		ll mid = sum[n/2];

		ll ans = Abs(mid);

		

		for(int i=1;i<n;i++)

			ans += Abs(sum[i]-mid);

		cout<<ans<<endl;

	}

	return 0;

}