BZOJ1045:[HAOI2008]糖果传递(贪心)

Description

有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。

Input

第一行一个正整数nn<=1'000'000，表示小朋友的个数．

接下来n行，每行一个整数ai，表示第i个小朋友得到的糖果的颗数．

Output

求使所有人获得均等糖果的最小代价。

Sample Input

4
1
2
5
4

Sample Output

4

Solution

emmm今天好颓啥都不想干怕不是省选后遗症

自己独立思考的能力还是不够，这也是我省选只能打暴力的原因……

还是要多练啊

https://www.cnblogs.com/CtrlCV/p/5626194.html 这个题解讲的还是很好的

让我这个有公式恐惧症的人都看懂了

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cstdlib>
#include<algorithm>
#define N (1000000+100)
using namespace std;

long long n,ave,ans,a[N],s[N];

int main()
{
    scanf("%lld",&n);
    for  (int i=1;i<=n;++i)
        scanf("%lld",&a[i]),ave+=a[i];
    ave/=n;
    for (int i=1;i<=n;++i)
        s[i]=s[i-1]+a[i]-ave;
    sort(s+1,s+n+1);
    for (int i=1;i<=n;++i)
        ans+=abs(s[i]-s[n/2+1]);
    printf("%lld",ans);
}