有n个小朋友坐成一圈,每人有a[i]个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为1。
求使所有人获得均等糖果的最小代价。
输入格式
第一行输入一个正整数n,表示小朋友的个数。
接下来n行,每行一个整数a[i],表示第i个小朋友初始得到的糖果的颗数。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤n≤1000000
数据保证一定有解。
输入样例:
4
1
2
5
4
输出样例:
4
设每个小朋友给出的糖果为xi,那么就可以求出每个人给出的糖果和第一个小朋友的关系,要求的是所有x绝对值加起来的最小值,xi-x1会是一个常数,就把所有常数排序取中位数,求出最小的代价。
#include<bits/stdc++.h>
using namespace std;
const int N=1000010;
int a[N];
int c[N]; int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
long long sum=0;
for(int i=1;i<=n;i++)
sum+=a[i];
int aver=sum/n;
c[n]=a[n]-aver;
for(int i=n-1;i>1;i--)
c[i]=c[i+1]+a[i]-aver;
c[1]=0;
sort(c+1,c+1+n);
long long res=0;
for(int i=1;i<=n;i++)
res+=abs(c[i]-c[(n+1)/2]);
cout<<res<<endl;
return 0;
}