这道题目需要一些代数推导,解法比较精妙
推导过程:
假设人从1开始到n编号,Ai表示初始时刻每个人手里面的金币数量,M表示金币的平均数,xi表示从i号位置拿出xi个金币给i-1号位置,这个值有正有负,负数表示反向的给于操作,那么可以得出下面的结论:
A1-x1+x2=M
A2-x2+x3=M
A3-x3+x4=M
........
A(n-1) - x(n-1) + x(1) = M
最后 一个算式是用前面的n个算式推导出来的,所以不能用这n个算式通过解方程的方法得出答案,但是可以把每一个xi都表示成x1的代数形式:
x1 = x1
x2 = x1-(A1-M)
x3 = x1-(A1-M)-(A2-M)
.........
假设(Ai-M)的n项累加和表示为Ci, 则有:
x1 = x1
x2 = x1 - C1
x3 = x1 - C2
........
xn = x1 - C(n-1)
题目要求移动的金币数量最少,即是|x1-0| + |x1-C1| + |x1-C2| + ........ + |x1-C(n-1)| 最小,也就把问题抽象成了另一个问题:
数轴上有n个点:0, C1, C2, ....., C(n-1),找一个点x1使得x1到这n个点的距离和最小,显然x1就是这n个点的中位数点,求出x1,问题就迎刃而解了。
下面是AC的代码:
#include <stdio.h>
#include <algorithm>
#define MAX_NUM 1000000
long long A[MAX_NUM];
long long C[MAX_NUM];
int cmp(const void* a, const void* b)
{
long long r;
r = *((long long*)a) - *((long long*)b);
if(r > 0)
return 1;
else if(0 == r)
return 0;
else
return -1;
}
int main(void)
{
int n, i;
long long M, middle, sum;
while(scanf("%d",&n) != EOF)
{
sum = 0;
for(i=0; i<n; i++)
{
scanf("%lld", A+i);
sum += A[i];
}
M = sum / n;
C[0] = 0;
for(i=1; i<n; i++)
{
C[i] = (A[i]-M) + C[i-1];
}
qsort((void*)C, n, sizeof(long long), cmp);
middle = C[n/2];
sum = 0;
for(i=0; i<n; i++)
{
sum += middle-C[i]>=0 ? middle-C[i] : C[i]-middle;
}
printf("%lld\n", sum);
}
return 0;
}