[HAOI2008]糖果传递

题目

传送门 to luogu

思路

我真是个弱菜，什么也不会 😢

不妨假设原有的糖果数量为 $a_1,a_2,a_3,\dots ,a_n$ 。

有一个很显然的性质，就是 $⟨i,i+1⟩$ 之间的糖果传递，不会是反反复复的—— $i$ 向 $i+1$ 给了 $a$ 个糖果，$i+1$ 向 $i$ 给了 $b$ 个糖果，相当于传递了 $|a-b|$ 个糖果。

所以我们假设 $x_i$ 是 $i$ 与 $i+1$ 之间的糖果传递数量。可正可负，正则为 $i$ 向 $i+1$ 传递，否则为 $i+1$ 向 $i$ 传递。

然后我们记平均数为 $k$ ，写出方程组

$$\{\begin{array}{l}{a}_1+x_n-x_1=k\\ a_2+x_1-x_2=k\\ \dots \\ a_i+x_{i-1}-x_i=k\\ \dots \\ a_n+x_{n-1}-x_n=k\end{array}$$

也就是说，$$\forall i\in (1,n],x_i=a_i+x_{i-1}-k$$

然后反复代换，直到边界 $x_2=a_2+x_1-k$ ，得到

$$\forall i\in (1,n],x_i=\sum _{j=2}^i{a}_j+x_1-(i-1)k$$

所以 $$cost=\sum _{i=1}^n|x_i|=\sum _{i=1}^n\left|x_1-[(i-1)k-\sum _{j=2}^i{a}_j]\right|$$

然后我们的目的就是求这个式子的最小值——

$$\sum _{i=1}^n|x-r_i|$$

最佳的 $x$ 是 { r } \{r\} {r} 的中位数。

代码

#include <cstdio>
#include <algorithm>
using namespace std;
int n, a[1000005];
long long r[1000005], k;
# define ABS(x) ((x)<0?-(x):(x))
int main(){
	scanf("%d",&n);
	for(int i=1; i<=n; ++i){
		scanf("%d",&a[i]);
		k += a[i];
	}
	k /= n, r[1] = 0;
	for(int i=2; i<=n; ++i)
		r[i] = r[i-1]-a[i]+k;
	sort(r+1,r+n+1);
	k = r[n>>1];
	long long ans = 0;
	for(int i=1; i<=n; ++i)
		ans += ABS(k-r[i]);
	printf("%lld\n",ans);
	return 0;
}