BZOJ3293_分金币_KEY

题目传送门

设x[i]表示i+1向i传的糖果数，x[n]表示1向n传的糖果数,a'=(a[1]+...a[N])/N
　　a[1]+x[1]−x[n]=a'
　　a[2]+x[2]−x[1]=a'
　　a[3]+x[3]−x[2]=a'
　　⋯⋯
　　a[n−1]+x[n−1]−x[n−2]=a'
　　a[n]+x[n]−x[n−1]=a'

把式子变形：
　　x[1]=a'−a[1]+x[n]
　　x[2]=a'−a[2]+x[1]=2∗a'−a[2]−a[1]+x[n]
　　x[3]=a'−a[3]+x[2]=3∗a'−a[3]−a[2]−a[1]+x[n]
　　⋯⋯
　　x[n−1]=a'−a[n−1]+x[n−2]=(n−1)∗a'−∑n−1i=1a[i]+x[n]
　　x[n]=n∗a'−∑ni=1a[i]+x[n]=0+x[n]

设s[i]=∑a[i]−i∗a'，则：
　　ans=∑∣x[i]∣ =∑∣s[i]−x[n] ∣
所以当x[n]为{s[1],s[2],...,s[n]}的中位数时答案最小

code:

/**************************************************************
    Problem: 3293
    User: yekehe
    Language: C++
    Result: Accepted
    Time:104 ms
    Memory:2480 kb
****************************************************************/

#include <cstdio>
#include <algorithm>
using namespace std;

char tc()
{
    static char fl[],*A=fl,*B=fl;
    return A==B&&(B=(A=fl)+fread(fl,,,stdin),A==B)?EOF:*A++;
}

long long read()
{
    char c;while(c=tc(),(c<''||c>'')&&c!='-');
    long long x=,y=;c=='-'?y=-:x=c-'';
    while(c=tc(),c>=''&&c<='')x=x*+c-'';
    return x*y;
}

const int MAXN=;

long long N,a[MAXN],sum[MAXN],w,ans,K;
int i;
int main()
{
//  freopen("x.txt","r",stdin);
    N=read();for(i=;i<=N;i++)a[i]=read(),w+=a[i];
    w=w/N;for(i=;i<=N;i++)sum[i]=sum[i-]+a[i]-w;
    sort(sum+,sum+N+);for(K=sum[N+>>]+sum[(N>>)+]>>,i=;i<=N;i++)ans+=abs(K-sum[i]);
    printf("%lld",ans);
    return ;
}