糖果传递(数学公式推导＋货仓选址问题)

题目

有 n 个小朋友坐成一圈，每人有 a[i] 个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为 1。
求使所有人获得均等糖果的最小代价。
输入格式

第一行输入一个正整数 n，表示小朋友的个数。
接下来 n 行，每行一个整数 a[i]，表示第 i 个小朋友初始得到的糖果的颗数。

输出格式

输出一个整数，表示最小代价。

数据范围

1≤n≤1000000,
0≤a[i]≤2×1e9,
数据保证一定有解。

输入样例：

4
1
2
5
4

输出样例：

4

思路:

总所周知代码越短越难,通过数学公式推导(离谱)将问题转化成了一个货仓选址问题,即数轴上给定N个点,在数轴上寻找一个点到其他所有点距离最短,如果N为奇数则就是最中间的点,如果N为偶数,则中间两点间任意一点都可(包括中间两点),公式推导问题转化的过程详见https://www.acwing.com/video/119/
这是看了视频以后自己推导的一遍，字很丑

代码:

#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
typedef long long LL;//怕这题数据爆int,干脆全部都用long long算了
const int N=1000010;
LL a[N],b[N];//b是用来存那一系列点的数组
LL res;
LL sum[N];//存前缀和的数组,求那一系列点要用到
int main()
{
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];
    }
    LL average=sum[n]/n;
    for(int i=2;i<=n;i++)
    b[i]=(n-i+1LL)*average-(sum[n]-sum[i-1]);
    sort(b+1,b+n+1);//对这些点排序
    for(int i=1;i<=n;i++)
    res+=abs(b[i]-b[(n+1)/2]);//b[(n+1)/2]最中间的点
    cout<<res<<endl;
    return 0;
}