分金币
Description
圆桌旁坐着n个人,每人有一定数量的金币,金币总数能被n整除。每个人可以给他左右相邻的人一些金币,最终使得每个人的金币数相等。你的任务是求出被转手的金币数量的最小值。比如,n=4,且4个人的金币数分别为1,2,5,4时,只需转移4枚金币(第3个人给第2个人两枚金币,第2个人和第4个人分别给第1 个人1枚金币)即可实现每人手中的金币数目相等。
输入包含多组数据。每组数据第一行为整数n(n<=1 000 000),以下n行每行为一个整数,按逆时针顺序给出每个人拥有的金币数。输入结束标志为文件结束符(EOF)。
对于每组数据,输出被转手金币数量的最小值。输入保证这个值在64位无符号整数范围内。
3
100
100
100
4
1
2
5
4
0
4
题解:
首先最终每个人的金币数量可以计算出来,它等于金币总数除以人数n即(sum/n)。
接下来用B来表示每个人最终拥有的金数
现在假设编号为 i 的人初始有Ai 枚金币,对于1号来说,他给了2号x1枚金币,还剩Ai -x1枚金币;但是n号给了他xn枚金币,所以还剩A1-x1+xn枚金币。所以A1-x1+x2=B。同理对于第2个人,有A2-x2+x1=B。依次类推。。。
最终得到n个方程,尝试用x1表示出其他的xi ,则本题就变成了单变量的极值问题。其实x1和xn代表的就是给出去的硬币和收回来的硬币。令 Ci=B-Ai;
对于第1个人,A1-x1+xn=B → x1=A1-B+xn=xn-C1(C1=B-A1)(D1=C1)
对于第2个人,A2-x2+x1=B → x2=A2-B+x1=A1+A2-2B+xn=xn-(C1+C2)(C2=B-A2)(D1=C1+C2)
对于第3个人,A3-x3+x2=B → x3=A3-B+x2=A1+A2+A3-3B+xn=xn-(C1+C2+C3)(C3=B-A3)(D3=C1+C2+C3)
。。。。
对于第n个人,An-xn+xn-1=B → xn=An-B+xn-1=A1+A2+A3+...An-nB+xn=xn-(C1+C2+C3+....Cn)(Cn=B-An)
Dn=(C1+C2...+Cn)
我们希望所有xi 的绝对值之和尽量小,即|xn-D1|+|xn-D2|+|xn-D3|+...+|xn|要最小。其实|xn-Di|的几何意义是数轴上的点xn到Di 的距离,所以问题变成了:给定数轴上n个点,找出一个到他们距离之和尽量小的点。也就是说我们是要寻找最xn这个点在哪里,我们才可以使总和最小,其实就是中位数的时候。
对于Di可以用前缀和求解
#include <stdio.h>
#include<algorithm>
using namespace std;
int a[1000010],b[1000010];
int main()
{
int sum,ave;
int n;
while(scanf("%d",&n)!=EOF)
{
sum=0;
int i;
for(i=1; i<=n; i++)
{
scanf("%d",&a[i]);
sum+=a[i];
}
ave=sum/n;
b[0]=0;
for(i=1; i<n; i++)
b[i]=b[i-1]+a[i]-ave;
sort(b,b+n);
int x=b[n/2],ans=0;
for(i=0; i<n; i++)
ans+=abs(x-b[i]);
printf("%d\n",ans);
}
return 0;
}