Description
有n个小朋友坐成一圈,每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。
Input
第一行一个正整数n<=987654321,表示小朋友的个数.接下来n行,每行一个整数ai,表示第i个小朋友得到的糖果的颗数.
Output
求使所有人获得均等糖果的最小代价。
Sample Input
4
1
2
5
4
Sample Output
4
题解
详见白书《训练指南》P5
代码
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
#define N 1000010
typedef long long ll;
ll a[N],c[N],tot,m,n;
int main()
{
scanf("%lld",&n);
for(int i = 1;i <= n;i++)
{
scanf("%lld",&a[i]);
tot += a[i];
}
m = tot / n;
c[0] = 0;
for(int i = 1;i < n;i++) c[i] = c[i-1] + a[i] - m;
sort(c,c+n);
m = c[n/2]; tot = 0;
for(int i = 0;i < n;i++) tot += abs(m - c[i]);
printf("%lld\n",tot);
return 0;
}