bzoj 1045: [HAOI2008] 糖果传递 （数学相关）

题目描述

传送门

题目大意：　有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。求使所有人获得均等糖果的最小代价。

题解

设xi表i给i+1传递的糖果的数量

那么我们可以得到n个方程

xn−x1=a¯−a1

x1−x2=a¯−a2

x2−x3=a¯−a3

⋯

xn−1−xn=a¯−an

那么对于方程做前缀和就能得到n个新的方程

xn−xi=∑ij=1a¯−aj

对于上面的方程进行变形，就能表示出xi

xi=xn−∑ij=1a¯−aj

我们的目标就是要最小化ans=∑ni=1|xi|=∑ni=1(xn−∑ij=1a¯−aj)

对于xn的取值，可以是不超过总和的任意值，显然所有的前缀和一定都在合法的范围之内。

那么我们只要让xn取到xn−∑ij=1a¯−aj中的中位数即可。

代码

#include<iostream>
#include<cstring>
#include<algorithm>
#include<cstdio>
#include<cmath>
#define N 1000003
#define LL long long
using namespace std;
LL a
,sum;
int n;
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++) scanf("%lld",&a[i]),sum+=a[i];
sum/=n;
a[1]=sum-a[1];
for (int i=2;i<=n;i++){
int t=sum-a[i];
a[i]=a[i-1]+t;
}
sort(a+1,a+n+1);
LL t=a[n/2+1]; LL ans=0;
for (int i=1;i<=n;i++) ans+=abs(t-a[i]);
printf("%lld\n",ans);
}