有n个小朋友坐成一圈,每人有a[i]个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为1。
求使所有人获得均等糖果的最小代价。
输入格式
第一行输入一个正整数n,表示小朋友的个数。
接下来n行,每行一个整数a[i],表示第i个小朋友初始得到的糖果的颗数。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤n≤1000000
数据保证一定有解。
输入样例:
4
1
2
5
4
输出样例:
4
```cpp
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int N = 1000015;
typedef long long ll;
int n;
ll c[N],p[N],mid,res,pa,sum;
int main()
{
cin >> n;
for(int i = 1; i <= n; i++)
{
scanf("%lld",&p[i]);
sum += p[i];
}
pa = sum / n;
c[n] = pa - p[1];
for(int i = n - 1; i > 0; i--) c[i] = c[i+1] + pa - p[i+1];
sort(c+1,c+n+1);
if(n % 2) mid = c[n/2 +1];
else mid = (c[n/2] + c[n/2+1]) / 2;
for(int i = 1; i <= n; i++) res += abs(c[i] - mid);
cout << res << endl;
return 0;
}