bzoj 1045 [HAOI2008] 糖果传递

1045: [HAOI2008] 糖果传递

Time Limit: 10 Sec   Memory Limit: 162 MB
Submit: 3155   Solved: 1439
[ Submit][ Status][ Discuss]

Description

　　有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。

Input

　　第一行一个正整数n<=987654321，表示小朋友的个数．接下来n行，每行一个整数ai，表示第i个小朋友得到的
糖果的颗数．

Output

　　求使所有人获得均等糖果的最小代价。

Sample Input

4
1
2
5
4

Sample Output

4

HINT

Source

好机智的一道题。。。http://hzwer.com/2656.html 这篇博客讲的很详细

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int n,i,a[1100000],b[1100000];
int main()
{
	while(scanf("%d",&n)!=EOF)
	{
		__int64 sum=0;
		for(i=1;i<=n;i++)
		{
			scanf("%d",&a[i]);
			sum+=a[i];
		}
		int avg=sum/n;
		for(i=2;i<=n;i++)
		{
			b[i]=b[i-1]+a[i]-avg;
		}
		sort(b+1,b+n+1);
		__int64 ans=0;
		int mid=b[n/2+1];
		for(i=1;i<=n;i++)
		ans+=abs(b[i]-mid);
		printf("%I64d\n",ans);
	}
	return 0;
}