有n个小朋友坐成一圈,每人有a[i]个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为1。
求使所有人获得均等糖果的最小代价。
输入格式
第一行输入一个正整数n,表示小朋友的个数。
接下来n行,每行一个整数a[i],表示第i个小朋友初始得到的糖果的颗数。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤n≤1000000
数据保证一定有解。
我们根据输入输出样例数据,进行推导
代码就比较简单了,推导过程比较难
#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
using namespace std;
const int N=1e6+10;
long long int st[N];
long long int cnt[N];
long long int c[N];
int main()
{
int n;
long long int ave=0;
cin>>n;
for(int i=1;i<=n;i++)
{
scanf("%lld",&st[i]);
cnt[i]=cnt[i-1]+st[i];
ave+=st[i];
}
ave=ave/n;
c[1]=0;
for(int i=2;i<=n;i++)
c[i]=(i-1)*ave-cnt[i-1];
sort(c+1,c+n+1);
long long int ans=0;
for(int i=1;i<=n;i++)
ans+=abs(c[i]-c[(n+1)/2]);
cout<<ans;
return 0;
}