[bzoj1045][HAOI2008] 糖果传递【构造】

【题目链接】
　　http://www.lydsy.com/JudgeOnline/problem.php?id=1045
【题解】
　　记$X[i]$为从第$i$堆向第$i+1$堆转移的数量。（第$n$堆转移到$1$）
　　那么有$a[i]-X[i]+X[i-1]=average$
　　$X[i]=a[i]+X[i-1]-average$
　　移项得：$X[1]=a[1]+X[n]-avarage$
　　　　　　$X[2]=a[2]+X[1]-avarage$
　　　　　　$=a[2]+a[1]+X[n]-2*average$
　　　　$X[i]=\sum_{j=1}^{i}a[i]-X[n]-i*avarage$
　　记$S[i]=\sum_{j=1}^{i}a[i]-i*avarage$
　有$X[i]=S[i]-X[n]$
　$ans=\sum_{i=1}^{n}|X[n]-S[i]|$
　当$X[n]$为$S[i]$中位数的时候$ans$取到最小值。
　复杂度$O(nlog_{2}n)$

/* --------------
    user Vanisher
    problem bzoj-1045 
----------------*/
# include <bits/stdc++.h>
# define    ll      long long
# define    inf     0x3f3f3f3f
# define    N       1000100
using namespace std;
ll read(){
    ll tmp=0, fh=1; char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-') fh=-1; ch=getchar();}
    while (ch>='0'&&ch<='9'){tmp=tmp*10+ch-'0'; ch=getchar();}
    return tmp*fh;
}
ll a[N],cnt,n,s[N],ans,now;
int main(){
    n=read();
    for (ll i=1; i<=n; i++)
        a[i]=read(), cnt+=a[i];
    cnt=cnt/n;
    for (ll i=1; i<=n; i++)
        s[i]=a[i]-cnt+s[i-1];
    sort(s+1,s+n+1);
    now=s[(1+n)/2];
    for (ll i=1; i<=n; i++)
        ans=ans+abs(s[i]-now);
    printf("%lld\n",ans);
    return 0;
}

同[bzoj1465][bzoj3293]