糖果传递

糖果传递

有n个小朋友坐成一圈，每人有a[i]个糖果。

每人只能给左右两人传递糖果。

每人每次传递一个糖果代价为1。

求使所有人获得均等糖果的最小代价。

输入格式
第一行输入一个正整数n，表示小朋友的个数。

接下来n行，每行一个整数a[i]，表示第i个小朋友初始得到的糖果的颗数。

输出格式
输出一个整数，表示最小代价。

数据范围
1≤n≤10000001≤n≤1000000
输入样例：
4
1
2
5
4
输出样例：
4

#include<iostream>
#include<algorithm>
using namespace std;
#pragma G++ optimize (3)
#pragma C++ optimize (3)
typedef long long ll;
const int maxn=1000000+10;
int f[maxn];
ll sum,num,a[maxn];
int n;
int main(){
	ios::sync_with_stdio(false);
        cin>>n;
        for(int i=1;i<=n;i++)
        {
                cin>>f[i];
                sum+=f[i];
        }
        sum/=n;
        for(int i=2;i<=n;i++)
        {
                a[i]+=a[i-1]-sum+f[i];
        }
        sort(a+1,a+1+n);
        for(int i=1;i<=n;i++)
        {
                num+=abs(a[i]-a[n/2+1]);
        }
        cout<<num<<endl;
        return 0;
}