1045: [HAOI2008] 糖果传递题解

题目链接

分析

这真的是省选题？？？
我们可先计算出每人最后所得糖果，然后计算前缀和，选择中位数，，每个数的前缀和减去中位数，这样就可以使代价最小，感觉水的一批。
时间：2600ms。
上代码

#include<bits/stdc++.h>
#define ll long long
using namespace std;
ll n,al[1000010],sum[1000010],ans,tot,num;
int main() {
    scanf("%lld",&n);
    for(int i=1;i<=n;i++) {
        scanf("%lld",&al[i]);
        tot+=al[i];
    }
    tot/=n;
    for(int i=1;i<=n;i++)
        al[i]-=tot,sum[i]=sum[i-1]+al[i];
    sort(sum+1,sum+n+1);
    num=sum[(n+1)/2];
    for(int i=1;i<=n;i++)
        ans=ans+abs(sum[i]-num);
    printf("%lld\n",ans);
    return 0;
}