BZOJ 1045: [HAOI2008] 糖果传递【数学题】

1045: [HAOI2008] 糖果传递

Time Limit: 10 Sec Memory Limit: 162 MB

Description

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

Input

第一行一个正整数nn<=1’000’000，表示小朋友的个数．
接下来n行，每行一个整数ai，表示第i个小朋友得到的糖果的颗数．

Output

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

Sample Input

4
1
2
5
4

Sample Output

4

题解

其实这题就是货物搬运。


完美解决

代码如下：

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int n,a[1000005];
long long ans,sum,b[1000005];
int read(){
    int ret=0,f=1,ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-') f=-f;ch=getchar();}
    while(ch>='0'&&ch<='9') ret=ret*10+ch-48,ch=getchar();
    return f*ret;
}
int _abs(int x){return x<0?-x:x;}
int main(){
    n=read();
    for(int i=1;i<=n;i++) sum+=a[i]=read();
    sum/=n;
    for(int i=1;i<=n;i++) b[i]=b[i-1]+a[i]-sum;
    sort(b+1,b+1+n);
    long long mid=b[(n+1)>>1];
    for(int i=1;i<=n;i++) ans+=_abs(b[i]-mid);
    printf("%lld\n",ans);
    return 0;
}