2-7 均等笔(本质上就是糖果传递)c++

emmm…这一题与糖果传递的区别就是把糖果改成了笔，类似的题目还有仓库选址。

题目：

n个人围成一圈，每人有ai支笔。每人可以向左右相邻的人传递笔，每人每次传递一支笔消耗的能量为1。求使所有人获得均等数量的笔的最小能量。

输入格式:
第一行一个整数n ，表示人的个数（30%的数据，n<=1000；100%的数据，n<=1e6）。
接下来n行，每行一个整数 ai。

输出格式:
输出一个整数，表示使所有人获得均等笔的最小能量。（答案保证可以用64位有符号整数存储）

输入样例:
4
1
2
5
4

输出样例:
4

（这里直接贴出我当时解题报告的分析）

#include<iostream>
#include<cmath>
#include<algorithm>
using namespace std;
typedef long long ll;
int a[1000006],c[1000006]={};
int main()
{
	ll n,sum=0,ave,num=0,mid;
	cin>>n;
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
		sum+=a[i];
	}	
	ave=sum/n;
	for(int i=1;i<n;i++)
		c[i]=c[i-1]+a[i]-ave;	
	sort(c+1,c+n);
	mid=c[n/2];
	for(int i=1;i<n;i++)
		num+=abs(mid-c[i]);
	num+=abs(mid);
	cout<<num;
	return 0;
}