【中位数】BZOJ1045(HAOI2008)[糖果传递]题解

题目概述

有 $n$ 个JZ坐成一圈，每人有 $a_i$ 个神犇值。每人只能给左右两人传递神犇值。每人每次传递一个神犇值代价为 $1$ 。用最少的代价使 $n$ 个JZ神犇值一样，以便虐人。

解题报告

远古省选题……被各种改题意，貌似已经烂大街了？

假设第 $i$ 个人向 $i+1$ 传递了 $x_i$ （负数表示 $i+1$ 向 $i$ 传递），令 $ave={\sum_{i=1}^{n}a_i\over n}$ ，那么：

$$x_n-x_1=ave-a_1\\ x_1-x_2=ave-a_2\\ \cdots\\ x_{n-1}-x_n=ave-a_n$$
令 $c_i=\sum_{i=1}^{n}ave-a_i$ ，累加后得到：
$$x_n-x_1=c_1\\ x_n-x_2=c_2\\ \cdots\\ x_n-x_{n}=c_n\\ ans=\sum_{i=1}^{n}|x_i|=\sum_{i=1}^{n}|x_n-c_i|$$
也就是说答案转化到了一个变量身上，这就比较好处理了。观察这个式子，我们可以认为这是 $x_n$ 与 $c_i$ 的距离之和，最小值在 $x_n$ 在 $\{c_n\}$ 的中位数取到。

示例程序

#include<cstdio>
#include<cctype>
#include<algorithm>
using namespace std;
typedef long long LL;
const int maxn=1000000;

int n,a[maxn+5];LL ave,sum[maxn+5],ans;

#define Eoln(x) ((x)==10||(x)==13||(x)==EOF)
inline char readc(){
    static char buf[100000],*l=buf,*r=buf;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF;return *l++;
}
inline int readi(int &x){
    int tot=0,f=1;char ch=readc(),lst='+';
    while (!isdigit(ch)) {if (ch==EOF) return EOF;lst=ch;ch=readc();}
    if (lst=='-') f=-f;
    while (isdigit(ch)) tot=(tot<<3)+(tot<<1)+(ch^48),ch=readc();
    return x=tot*f,Eoln(ch);
}
inline LL Abs(LL x) {if (x<0) return -x;return x;}
int main(){
    freopen("program.in","r",stdin);
    freopen("program.out","w",stdout);
    readi(n);for (int i=1;i<=n;i++) readi(a[i]),ave+=a[i];
    ave/=n;for (int i=1;i<=n;i++) sum[i]=sum[i-1]+ave-a[i];
    sort(sum+1,sum+1+n);LL mid=sum[n+1>>1];
    for (int i=1;i<=n;i++) ans+=Abs(sum[i]-mid);
    return printf("%lld\n",ans),0;
}