糖果传递(贪心,推公式)

题目链接:https://www.acwing.com/problem/content/124/

题目
有 $n$ 个小朋友坐成一圈，每人有 $a[i]$ 个糖果。

每人只能给左右两人传递糖果。

每人每次传递一个糖果代价为 $1$。

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

输入格式
第一行输入一个正整数 $n$，表示小朋友的个数。

接下来 $n$ 行，每行一个整数 $a[i]$，表示第 $i$ 个小朋友初始得到的糖果的颗数。

输出格式
输出一个整数，表示最小代价。

数据范围
$1\le n\le 1000000,$
$0\le a[i]\le 2\times 10^9,$
数据保证一定有解。

输入样例：

4
1
2
5
4

输出样例：

4

思路:

如图所示（代价即交换次数）

假设$x_i$表示$A_{i-1}$传递到$A_i$的代价为$x_i$
假设$-x_i$表示$A_i$传递到$A_{i-1}$的代价为$x_i$
则总代价为$|x_1|+|x_2|+...+|x_n|$

下面研究每一个位置，平均数为avg
对于$A_1$，有$A_1-x_1+x_n=avg$
对于$A_2$，有$A_2-x_2+x_1=avg$
对于$A_3$，有$A_3-x_3+x_2=avg$
$...$
对于$A_{n-1}$，有$A_{n-1}-x_{n-1}+x_{n-2}=avg$
对于$A_n$，有$A_n-x_n+x_{n-1}=avg$

移项可得
$A_1-avg=x_1-x_n$
$A_2-avg=x_2-x_1$
$A_3-avg=x_3-x_2$
$...$
$A_{n-2}-avg=x_{n-2}-x_{n-3}$
$A_{n-1}-avg=x_{n-1}-x_{n-2}$
$A_n-avg=x_n-x_{n-1}$

把最后两项相加，有
$A_n+A_{n-1}-2avg-x_n=-x_{n-2}$

把最后三项相加，有
$A_n+A_{n-1}+A_{n-2}-3avg-x_n=-x_{n-3}$

则
$0-x_n=-x_n$
$A_n-avg-x_n=-x_{n-1}$
$A_n+A_{n-1}-2avg-x_n=-x_{n-2}$
$A_n+A_{n-1}+A_{n-2}-3avg-x_n=-x_{n-3}$
$A_n+A_{n-1}+A_{n-2}+A_{n-3}-4avg-x_n=-x_{n-4}$
$...$
$A_n+A_{n-1}+...+A_2-(n-1)avg-x_n=-x_1$

令
$C_1=A_n-avg$
$C_2=A_n+A_{n-1}-2avg$
$C_3=A_n+A_{n-1}+A_{n-2}-3avg$
$...$
$C_{n-1}=A_n+A_{n-1}+...+A_2-(n-1)avg$

则总代价可以表示为
$|C_1-x_n|+|C_2-x_n|+|C_3-x_n|+...+|C_{n-1}-x_n|+|x_n|$

由货仓选址可知,取中位数可以使不等式的值最小

另外$C_i=C_{i-1}+A_i-avg$

AC代码

#include <iostream>
#include <cstdio>
#include <cmath>
#include <algorithm>

using namespace std;

const int N = 1000010;

typedef long long LL;

int n;
LL arr[N],c[N];
LL all,avg,ans;

int main()
{
    scanf("%d",&n);
    for(int i = 0; i < n; i++)
    {
        scanf("%lld",&arr[i]);
        all += arr[i];
    }
    avg = all / n;
    c[0] = 0;
    // for(int i = 1; i < n; i++) c[i] = c[i-1] + avg - arr[i];
    // 上面这个递推式也行
    for(int i = 1; i < n; i++) c[i] = c[i-1] - avg + arr[i];
    sort(c,c+n);
    avg = c[n/2];
    for(int i = 0; i < n; i++) ans += abs(c[i] - avg);
    printf("%lld\n",ans);
    return 0;
}