Uva 11300
题目大意为有n个人,围成一个圈,每个人有ai个金币,且所有人的金币总和可以被n整除,每个人可以给自己左边和右边的人金币,要求最后所有人的金币相等。
我们假设每个人只会给自己左边的人金币,并且只会从自己右边的人那里获得金币,因为如果即从左边获得又从右边获得可以转化为只从一边获得,则对于第i个人有ai-xi+xi+1=avg,假设第一个人支出的金币为x1,则第二个人支出的金币x2=x1-a1+avg,第三个人的为x3=avg*2+x1-a1-a2,递推得第n个人获得的金币数为xn=an-1+an-2……+a2+x1-avg*(n-1),设**an-1+an-2……+a2-avg*(n-1)=-c[i]**因此只需要递推得到c[i]即可。
代码如下
for(int i=2;i<=n;i++)
c[i]=c[i-1]+arr[i]-avg;
故该问题转化为求|x1|+|x1-c1|+|x1-c2|+……+|x1-cn| 的值最小,于是相当于在数轴上找到一个点,让其他n个点到他的距离最小,那么这个点应该是这些数中的中位数,证明如下:
假设这个点为P,且P不是这n个点中的中位数,则在他的左边有a个点,右边有n-a-1个点,且a!=n-a-1,那么如果将这个点向左移动一段距离x,如果左边的点的距离到P的距离和为d,右边点到P的距离和为d’,那么
假设d>d’,则在移动之后答案变得更优了,所以应该找到的点在不管向左移还是向右移答案都应该变得更差,即左边和右边的距离相等,故这个点为中位数的点。
下面为完整代码:
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<stdlib.h>
#define LL long long
using namespace std;
LL n,c[1000100],arr[1000100];
bool cmp(int a,int b){
return a<b;
}
int main()
{
while(scanf("%lld",&n)!=EOF){
if(n==0){
printf("0\n");
continue;
}
memset(c,0,sizeof(c));
LL ans=0;
LL sum=0,avg=0;
for(int i=1;i<=n;i++){
scanf("%lld",&arr[i]);
sum+=arr[i];
}
avg=sum/n;
for(int i=2;i<=n;i++)
c[i]=c[i-1]+arr[i]-avg;
sort(c+1,c+n+1,cmp);
LL x1;
x1=c[n/2+1];
for(int i=1;i<=n;i++)
ans+=abs(x1-c[i]);
printf("%lld\n",ans);
}
return 0;
}