122. 糖果传递（均分纸牌加强版）

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

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

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

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

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

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

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

数据范围
1≤n≤1000000
数据保证一定有解。

输入样例：

4
1
2
5
4

输出样例：

4

思路分析

设糖果平均数为a，a_i代表第i堆糖果的个数，x_i代表第i堆糖果向第i+1堆糖果（第n堆向第1堆传递）传递的糖果数目。温馨提示：x_i代表是传递的差值（有很多种传递方案的）。
则
a₁ - x₁ + x_n= a
a₂ - x₂ + x₁ = a (含义：第1堆糖果向第2堆糖果传递了x₁颗，第2堆糖果向第3堆糖果传递了x₂颗，传递后第2堆糖果的数目等于糖果的平均数a)
…
a_n-1 - x_n-1 + x_n-2 = a
a_n - x_n + x_n-1 = a
∵有上面方程组等价于
x₁ - x_n=a₁ - a
x₂ - x₁ = a₂ - a
…
x_n-1 - x_n-2 = a_n-1 - a
x_n + x_n-1 =a_n - a
上面所有式子相加得0=0恒成立，所以上式中至少有一个多余方程。
∴令x₁为自由变量，用x₁表示其他的变量。
x₁ = x₁ + 0
x₂= x₁ + a₂ - a
x₃= x₁ + a₃ + a₂ - 2a（由x₃= x₂ + a₃ - a转化而来）
…
x_n-1= x₁ + a_n-1 + a_n-2 + … + a₂ - （n-2）a
x_n =x₁ + a_n + a_n-1 + a_n-2 + … + a₂ - （n-1）a
因为加法不好观察规律，所以将上式化为减法
x₁ = x₁ - 0
x₂= x₁ - （a - a₂）
x₃= x₁ -（2a - a₂ - a₃）
…
x_n =x₁ - （（n-1）a - a₂- a₃ + a₄ + … + a_n）
设 x₁之后的部分为C_i
x₁ = x₁ - C₁
x₂= x₁ - C₂
x₃= x₁ - C₃
…
x_n =x₁ - C_n
我们要求的目标是| x₁ | + | x₂ | + … +| x_n |的最小值
即求| x₁ - C₁ | + | x₁ - C₂ | + … +| x₁ - C_n |的最小值
∵| x₁ - C_n | = | -（C_n - x₁） | = | C_n - x₁ |
∴ | C_n - x₁ | + | x₁ - C₁ | ≥ |C_n - C₁ |(温馨提示由| x | + | y | ≥ | x + y | 推导而来)
前后两两分组，取中位数即可取得最小值。（与货仓选址（邮递员问题）思路类似，看不懂的可以先看货仓选址）。
证明能取得最优解
∵ x₁ 取的是 C_i（1≤i≤n）中位数。
∴x₁ - C_i肯定有一部分大于等于0，一部分小于等于0。
则必存在情况①x_i ≥ 0且x_i+1≤ 0 （对于第i+1堆糖果而言，都是别人给他或者他不用给别人），②x_i ≤ 0且x_i+1 ≥ 0 （对于第i+1堆糖果而言，都是他给别人或者他不用给别人）这两种情况先（给/收）左边这家还是先（给/收）右边这家，顺序无影响。所以能将环断开成链从而退化成均分纸牌问题（与均分纸牌思路类似，看不懂的可以先看均分纸牌）。注意点：别人给他，他再给下一家，这种顺序是有影响的。别人要先给他，他才够糖果给别人。
具体方案递归策略：
①x_i ≥ 0且x_i+1≤ 0 时先递归别人，最后别人再把糖果传递给他。
②x_i ≤ 0且x_i+1 ≥ 0 时他先把糖果传递给别人，最后再递归计算别人。

代码

#include<cstdio>
#include<algorithm>
#include<iostream>
using namespace std;
typedef long long LL;
const int N=1000005;
LL s[N],c[N];
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=n;i++){
        scanf("%lld",&s[i]);
        s[i]+=s[i-1];//s[i]表示a[i]的前i项和
    }
    LL avg = s[n]/n;//求a[i]的平均数，因为s[n]是a的前n项和,所以直接除n即可
    for(int i=2;i<=n;i++){//c[1]=0,所以直接从2开始计算即可
        c[i] = (i-1)*avg - (s[i]-s[1]);//求c[i]=(i-1)*avg-(a[2]+a[3]+...+a[i])
    }//参数：起始下标，选择下标，结束下标 
    nth_element(c+1,c+(n+1)/2,c+n+1);//求c数组的中位数，注意：下标从1开始
    LL md = c[(n+1)/2];//将中位数赋值给变量md
    LL res = 0;
    for(int i =1;i<=n;i++){//统计差值，思路与货仓选址类似
        res += abs(c[i]-md);
    }
    cout<<res<<endl;
    return 0;
}