P2512 [HAOI2008]糖果传递(环形“均分纸牌“可链化，前缀和数组)

P2512 [HAOI2008]糖果传递

均分纸牌题解

题意：$n$个人，每个人有$A_i$颗糖果，第$i$人给第$i-1$人$X_i$颗糖果(第一人给第$n$人，$X_i$可正可负，负数表示第$i-1$个给第i人)，设平均值为$ave$，每给一颗糖果记为一次操作，问最少操作次数使n个人的糖果数量最少。

  ~  
  ~  

参考大佬博客F_J

主要证明(复述):

$ave=A_1+X_2-X_1$
$ave=A_2+X_3-X_2$
……
$ave=A_n+X_1-X_n$
$==>$

$X_2=X_1-(A_1-ave)$
$X_3=X_1-(A_1+A_2-ave\ast 2)$
……
$X_n=X_1-(A_1+A_2......A_n-ave\ast (n-1))$

令$S_i={\sum }_{i=1}^{n}A[i]-ave$
$==>$

$X_2=X_1-S_1$
$X_3=X_1-S_2$
……
$X_n=X_1-S_n$

操作次数就是：

$ans={\sum }_{i=1}^n{X}_i={\sum }_{i=1}^n\mid X_1-S_i\mid$

那么原问题变成了求 X1 的值使得上述求和最小，上述求和可以看作是求点到直线的距离。
先将 S 数组从小到大排序。
如果 n 为奇数，那么 X1=Sn/2+1
如果 n 为偶数，那么 X1∈[Sn/2,Sn/2+1]
可以发现，不论 n 为奇数还是偶数，X1 都可以取到S 数组中的某个值使得求和最小，因此存在某个 Xi=X1−Si=0，即有两个人之间没有纸牌传递，因此这种解法实际上证明了环形均分纸牌问题的可链化。

结论：
我们取数组$b[i]=A_i-ave$的前缀和数组$sum[i]$的中值为$X_1$，求出最小操作次数。

#include<bits/stdc++.h>
using namespace std;
const int N = 1e6+10;
typedef long long ll;
ll s[N], sum[N];
int main() {
    ll n, ave = 0;
    cin >> n;
    for(int i=1; i<=n; i++) {
        cin >> s[i]; ave += s[i];
    }
    ave /= n;
    for(int i=1; i<=n; i++) {
        s[i] -= ave; sum[i] = sum[i-1] + s[i];
    }
    sort(sum+1, sum+1+n);
    ll mid = (1+n)/2, ans = 0;
    for(int i=1; i<=n; i++) {
         ans += abs(sum[mid] - sum[i]);
    }
    cout << ans;
    return 0;
}