题目
有n个小朋友坐成一圈,每人有a[i]个糖果。
每人只能给左右两人传递糖果。
每人每次传递一个糖果代价为1。
求使所有人获得均等糖果的最小代价。
输入格式
第一行输入一个正整数n,表示小朋友的个数。
接下来n 行,每行一个整数a[i],表示第i 个小朋友初始得到的糖果的颗数。
输出格式
输出一个整数,表示最小代价。
数据范围
1≤n≤1000000,
0≤a[i]≤2×10的9次方,
数据保证一定有解。
输入样例:
41254
输出样例:
4
思路
- 糖果不会凭空增加或者消失;本题只考虑一定有解的情况,也就是总数一定能够整除n;不会出现环形给的情况:a->b->c->a;
- 数据范围是1e6,所以要做到nlogn或者是O(n)的情况;
- 建立数学模型:
- xi可正可负;
- 目标:使所有xi的绝对值相加之和最小
- 限制:
- n-1个独立的方程,n个未知数,可以用某一个的x来表示其他的x
- 目标函数转化为:(类似上一题的货物运输:求中位数+求Ci)
代码
n = int(input())a=[int(input()) for _ in range(n)]# print(l1)c=[0 for _ in range(n)]avg=sum(a)//nfor i in range(n-1,0,-1):c[i-1] = c[i]+avg-a[i] # 注意下标范围是从0开始的,题解下标从1开始c.sort()res=0for i in range(n):res+=abs(c[i]-c[(n-1)//2])print(res)