#10010. 「一本通 1.1 练习 6」糖果传递

题目描述
原题来自：HAOI 2008
有 n 个小朋友坐成一圈，每人有 ai 颗糖果 每人只能给左右两人传递糖果。每人每次传递一颗糖果的代价为 1 求使所有人获得均等糖果的最小代价
输入格式
第一行有一个整数 n ，表示小朋友个数
在接下来 n 行中，每行一个整数 ai
输出格式
输出使所有人获得均等糖果的最小代价。
提示
不开long long见祖宗（bushi
AC代码

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
long long a[1000005],b[1000005];
int main()
{
	long long n;
	cin>>n;
	long long s=0;
	for(long long i=1;i<=n;i++){
		cin>>a[i];
		s+=a[i];
	}
	s/=n;
	for(long long i=1;i<=n;i++){
		b[i]=b[i-1]-a[i]+s;//b[i-1]-a[i]+s
	}
	sort(b+1,b+1+n);
	long long sum=0;
	long long mid=b[(1+n)/2];
	for(long long i=1;i<=n;i++){
		sum+=abs(b[i]-mid);//最小代价
	}
	cout<<sum;
	return 0;
}