题目背景
在新的一年,mxj给俱乐部的孩子们准备了一堆的礼物,它生成了一个随机数列表aa,使得第i个人会得到a[i]件礼物。
但是,因为某些人得到的礼物太多,为了公平起见,mxj建议大家把礼物平均分配。
题目描述
大家围成了一个圈,每个人可以向他左边或者右边的人传递礼物,问,至少传递多少件礼物,才能让每个人得到的礼物件数相同呢?(保证礼物总数是人数n的倍数)
输入格式:
第一行输入一个正整数n,表示俱乐部成员的个数。
接下来有n行,每行一个整数a[i],表示第i个成员初始得到的礼物的件数。
输出格式:
输出一个整数,表示至少需要传递多少件礼物。
输入样例#1:
4
1
2
5
4
输出样例#1:
4
说明
1≤n≤1000000
0≤a[i]≤109
解题思路
简单的贪心问题,不多bb。
代码
#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL a[1000005];
LL l[1000005]={0};
int main()
{
LL n,sum=0;
scanf("%lld",&n);
for(int i=0; i<n; i++)
{
scanf("%lld",&a[i]);
sum+=a[i];
}
sum/=n;
LL ans=0;
for(int i=0;i<n;i++)
a[i]-=sum;
for(int i=0;i<n;i++)
l[i]=l[i-1]+a[i];
sort(l,l+n);
for(int i=0;i<n;i++)
ans+=abs(l[n/2]-l[i]);
printf("%lld\n",ans);
return 0;
}