BZOJ 1045 HAOI 2008 糖果传递 数学推导

好像Chrome支持MathJax有点问题。。Microsoft Edge一点事没有。。
因为一个一个小朋友传递，所以只用考虑两相邻两个小朋友间的关系。
令$x_i$表示小朋友$i-1$给$i$的糖果数，$v=\frac{1}{n}\sum a_i$，那么显然有：

$$ans=\sum \left|x_i\right|$$
而且对于每个小朋友有：
$$v=a_i+x_i-x_{i+1}$$
因此
$$x_{i+1}=x_i+a_i-v$$
因此， $\{x_i\}$之间存在一个关系，使得某个 $x_i$定下来了以后，其他 $x_j$都确定了，因而答案 $ans$也确定了。
假设我们固定 $x_1$，那么对于其他数字，如
$$\begin{aligned}x_2&=x_1+a_1-v\\ x_3&=x_2+a_2-v=x_1+a_1-v+a_2-v\\&\cdots\end{aligned}$$
我们设 $c_0=0,c_i=c_{i-1}-a_i+v$，那么有
$$x_i=x_1-c_i$$
答案有
$$ans=\left|x_1-c_0\right|+\left|x_1-c_1\right|+\left|x_1-c_2\right|+\cdots+\left|x_1-c_{n-1}\right|$$
发现这个式子表示的就是 $x_1$在数轴上与 $c_i$的距离和，要使距离和最小，就要使 $x_1=c_i$的中位数，问题就解决了。

发的时候用I64d跪了。。。

#include <cstdio>
#include <cstdlib>
#include <algorithm>
using namespace std;
typedef long long ll;
#define FOR(i,j,k) for(i=j;i<=k;i++)
const int N = 10000005;

ll read() {
    ll s = 0, f = 1; char ch = getchar();
    for (; ch < '0' || ch > '9'; ch = getchar()) if (ch == '-') f = -1;
    for (; '0' <= ch && ch <= '9'; ch = getchar()) s = s * 10 + ch - '0';
    return s * f;
}

int main() {
    static int a[N], s[N];
    ll n = read(), i, x, tot = 0, avg, mid, ans;
    s[0] = 0;
    FOR(i,1,n) a[i] = read(), tot += a[i];
    avg = tot / n;
    FOR(i,1,n-1) s[i] = s[i - 1] + a[i] - avg;
    sort(s + 1, s + n);
    mid = s[n / 2];
    ans = abs(mid);
    FOR(i,1,n-1) ans += abs(s[i] - mid);
    printf("%lld\n", ans);
    return 0;
}

1045: [HAOI2008] 糖果传递

Time Limit: 10 Sec Memory Limit: 162 MB
Submit: 3047 Solved: 1378

Description

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

Input

小朋友个数n 下面n行 ai

Output

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

Sample Input

4
1
2
5
4

Sample Output

4

HINT

100% n<=987654321