BZOJ3293 [CQOI2011]分金币

Address

• BZOJ3293 洛谷P3156

Description

• 圆桌上坐着$n$个人，每人有一定数量的金币，金币总数能被$n$整除。每个人可以给他左右相邻的人一些金币，最终使得每个人的金币数目相等。你的任务是求出被转手的金币数量的最小值。

Input

• 第一行为整数$n(n \ge 3)$，以下$n$行每行一个正整数，按逆时针顺序给出每个人拥有的金币数。

Output

• 输出被转手金币数量的最小值。

Sample Input

• 4
  1
  2
  5
  4

Sample Output

• 4

Interpretation

• 设四个人编号为$1,2,3,4$。第3个人给第2个人2个金币（变成$1,4,3,4$），第2个人和第4个人分别给第1个人1个金币。

Hint

• $N \le 100000$，总金币数$\le 10^9$

Solution 数学

• 设$x_i$表示第$i - 1$个人给第$i$个人的金币数，记第$i$人拥有的金币数为$a_i$且平均数为$ave$
• 那么我们会得到$n$个形如$a_i + x_{i - 1} - x_i = ave$的方程，求最小的$\sum \limits_{i = 1}^n |x_i|$
• 移项得到$x_i = x_{i - 1} + a_i - ave$这一递推式，也就是说只要知道$x_1$就确定其余的$x_i$
• 考虑把$\sum \limits_{i = 1}^n x_i$表示成$\sum \limits_{i = 1}^n |x_1 - delete_i|$的形式，进一步地我们能够推出$delete_i = (i - 1) * ave - \sum \limits_{j = 2}^i a_j$，从而直接算出$delete_i$
• 那么$\sum \limits_{i = 1}^n |x_1 - delete_i|$的几何意义就是找到一个最优的点$x_1$使得它到$n$个点$delete_i$的距离之和最小
• 因此$x_1$毫无疑问是$delete_i$序列的中位数，算出其对应的答案即可

Code

#include <iostream>
#include <cstdio>
#include <cctype>
#include <algorithm>
#include <cstring>

using namespace std;

namespace INOUT
{
    const int S = 1 << 20;
    char frd[S], *hed = frd + S;
    const char *tal = hed;

    inline char nxtChar()
    {
        if (hed == tal)
            fread(frd, 1, S, stdin), hed = frd;
        return *hed++;
    }

    inline int get()
    {
        char ch; int res = 0; bool flag = false;
        while (!isdigit(ch = nxtChar()) && ch != '-');
        (ch == '-' ? flag = true : res = ch ^ 48);
        while (isdigit(ch = nxtChar()))
            res = res * 10 + ch - 48;
        return flag ? -res : res;
    }
};
using namespace INOUT;

const int N = 1e5 + 5;
int sum, n, ave, del[N]; long long Ans;

int main()
{
    n = get();
    for (int i = 1; i <= n; ++i)
        sum += get(), del[i] = -sum; 
    //del[i]都减去a[1]显然对结果没有影响 
    ave = sum / n; sum = 0;
    for (int i = 1; i <= n; ++i)
        del[i] += sum, sum += ave;
    sort(del + 1, del + n + 1);
    for (int i = 1, im = n >> 1; i <= im; ++i)
        Ans += (long long)del[n - i + 1] - del[i];
    cout << Ans << endl;
    return 0;
}