题目
有 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;
}