【公式】P2512 [HAOI2008]糖果传递

原题地址

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

输入格式
小朋友个数 $n$，下面 $n$ 行 $a_i$ 。

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

输入输出样例
输入 #1

4
1
2
5
4

输出 #1

4

说明/提示
对于 $100\%$ 的数据 $n\le 10^6$ 。

---

思路：

设每个人原来拥有糖果数量为 $A_i$ ，每次向右传递数量为 $X_i$ （向左为 $-X_i$ ），糖果平均数为 $avg$ 。

则有
$$原有的-向右传出去的+左边传进来的=平均数$$
即
$$A_i-X_i+X_{i+1}=avg$$
联立方程
$$\{\begin{array}{rl}{A}_1-X_1+X_2& =avg\\ A_2-X_2+X_3& =avg\\ \cdots & \\ A_{n-1}-X_{n-1}+X_n& =avg\\ A_n-X_n+X_1& =avg\end{array}$$
整理得
$$\{\begin{array}{rl}-X_1+X_2& =avg-A_1\\ -X_2+X_3& =avg-A_2\\ \cdots & \\ -X_{n-1}+X_n& =avg-A_{n-1}\\ X_1-X_n& =avg-A_n\end{array}$$
其增广矩阵

$$B=(A\beta )=\left[\begin{array}{ccccccc}-1& 1& 0& \cdots & 0& 0& ave-A_1\\ 0& -1& 1& \cdots & 0& 0& ave-A_2\\ ⋮& & \ddots & & & ⋮& ⋮\\ 0& 0& 0& \cdots & -1& 1& ave-A_{n-1}\\ 1& 0& 0& \cdots & 0& -1& ave-A_n\end{array}\right]$$
化为行最简形
$$\left[\begin{array}{ccccccc}1& 0& 0& \cdots & 0& -1& ({\sum }_{i=1}^{n-1}{A}_i)-(n-1)ave\\ 0& 1& 0& \cdots & 0& -1& ({\sum }_{i=2}^{n-1}{A}_i)-(n-2)ave\\ ⋮& & \ddots & & & ⋮& ⋮\\ 0& 0& 0& \cdots & 1& -1& A_{n-1}-ave\\ 0& 0& 0& \cdots & 0& 0& 0\end{array}\right]$$
得方程组的解为
$$\{\begin{array}{rlr}{X}_1& =X_n+(\sum _{i=1}^{n-1}{A}_i)& -(n-1)ave\\ X_2& =X_n+(\sum _{i=2}^{n-1}{A}_i)& -(n-2)ave\\ & ⋮& \\ X_{n-1}& =X_n+A_{n-1}& -ave\\ X_n& =X_n& \in \mathbb{R}\end{array}$$
现在要求 ${\sum }_{i=1}^n|X_i|$ 的最小值

观察方程组的解，发现所有项都有相同的变量 $X_n$ ，所以构造函数
$$X_x=X_n+f(x)$$
即
$$f(x)=\sum _{i=x}^{n-1}{A}_i-(n-x)ave$$
所以
$$\sum _{i=1}^n|X_i|=|X_n+f(1)|+\cdots +|X_n+f(n)|=\sum _{i=1}^n|X_n+f(i)|$$
而 ${\sum }_{i=1}^n|X_n+f(i)|$ 可以看作数轴上的点 $X_n$ 到点 $-f(i),i\in [1,n]$ 的距离之和

那么 $X_n$ 应取最中间的点（高中结论，函数图像为平底锅（ $n$ 为偶数）或尖底锅（ $n$ 为奇数））

于是我们只要找到 $-f(i),i\in [1,n]$ 的中位数，把他赋值给 $X_n$ 即可
将其代入${\sum }_{i=1}^n|X_n+f(i)|$ 即为本题答案

$PS:$ $f(x)$ 有递归求法 $f(i)=f(i+1)+A_i-ave$ ，从后往前循环即可，不要用 $f(x)={\sum }_{i=x}^{n-1}{A}_i-(n-x)ave$ ，后者会导致 $TLE$

---

$AC$ 代码

#include<bits/stdc++.h>
#define FOR(i,a,b) for(int i=(a);i<=(b);++i)
#define ROF(i,a,b) for(int i=(a);i>=(b);--i)
#define ll long long
using namespace std;

const int N = 1e6+1;
int n,a[N],f[N];
ll ave=0,ans=0,Xn;

int main(){
	cin>>n;
	FOR(i,1,n){
		cin>>a[i];
		ave+=a[i];
	}
	ave/=n;//由题意，一定能整除
	f[n]=0;
	ROF(i,n-1,1)
		f[i]=f[i+1]+a[i]-ave;
	sort(f+1,f+n+1);
	Xn=-f[(n+1)/2];
	FOR(i,1,n)
		ans+=abs(Xn+f[i]);
	cout<<ans;
	return 0;
}