有n个小朋友坐成一圈,每人有a[i]个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为1。
求使所有人获得均等糖果的最小代价。
输入格式
第一行输入一个正整数n,表示小朋友的个数。
接下来n行,每行一个整数a[i],表示第i个小朋友初始得到的糖果的颗数。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤n≤1000000
数据保证一定有解。
输入样例:
4
1
2
5
4
输出样例:
4
取这些数的中位数
#include<iostream>
#include<algorithm>
#define LL long long
using namespace std;
int n;
LL s[1000010],c[1000010],b;
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++)
{
scanf("%lld",&s[i]);
s[i]+=s[i-1];
}
b=s[n]/n;//平均值
int k=0;
for (int i = 1; i < n; i ++ ) c[k++] = i * b - s[i];
c[k++]=0;
sort(c,c+k);
LL res=0;
int t=c[k/2];
for (int i = 0; i < k; i ++ ){
res+=abs(t-c[i]);
}
printf("%lld\n", res);
return 0;
}