[bzoj1045][bzoj3293][数论]糖果传递/分金币

Description

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

Input

第一行一个正整数nn<=1’000’000，表示小朋友的个数． 接下来n行，每行一个整数ai，表示第i个小朋友得到的糖果的颗数．

Output

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

Sample Input

4

1

2

5

4

Sample Output

4

题解

设a[i]表示第i个人给了第i+1个人多少
设s[i]表示第i个人原本有多少个
容易看出最后每个人的糖果数都是平均数
可列方程
s[1]+a[n]-a[1]=avg
s[2]+a[1]-a[2]=avg
….
s[n]+a[n-1]-a[n]=avg
移项可得
a[1]=s[1]+a[n]-avg
a[2]=s[2]+a[1]-avg=s[2]+s[1]+a[n]-2*avg
a[3]=s[3]+a[2]-avg=s[3]+s[2]+s[1]+a[n]-3*avg
…
把a[n]延后，可发现前面项均为前缀和减去平均数形式，设他们为C[i]
问题可变为求sigma(a[n]-C[i])
C数组为定值，原问题仍可转化为在数轴上有n个点，选一个点使得这n个点到这个点的总距离最小
容易看出这个点是中位数

#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;
int n;
LL a[1110000],avg;
int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)scanf("%lld",&a[i]),avg+=a[i];
    avg/=n;
    for(int i=1;i<=n;i++)a[i]=a[i-1]+a[i]-avg;
    sort(a+1,a+1+n);
    int tmp=n/2+1;LL ans=0;
    for(int i=1;i<=n;i++)ans+=abs(a[tmp]-a[i]);
    printf("%lld\n",ans);
    return 0;
}