P2512 【一本通提高篇贪心】「一本通 1.1 练习 6」[HAOI2008]糖果传递

[HAOI2008]糖果传递

题目描述

有

n

n

n 个小朋友坐成一圈，每人有

a

i

a_i

ai​ 个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为

1

1

1。

输入格式

小朋友个数

n

n

n，下面

n

n

n 行

a

i

a_i

ai​。

输出格式

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

输入输出样例

样例输入1

4
1
2
5
4

样例输出1

4

说明 / 提示

对于

100

%

100\%

100% 的数据

n

≤

1

0

6

n\le 10^6

n≤106。

Code

#include <bits/stdc++.h>
#define int long long
using namespace std;
int n, a[1000005], c[1000005], t, s, ans;

int jdz(int x)
{
	return x < 0 ? -x : x;
}

signed main()
{
	cin >> n;
	for (int i(1); i <= n; ++i)
	{
		scanf("%d", &a[i]);
		s += a[i];
	}
	t = s / n;
	for (int i(2); i <= n; ++i)
		c[i] = c[i - 1] + a[i] - t;
	sort(c + 1, c + n + 1);
	for (int i(1); i <= n; ++i)
		ans += jdz(c[i] - c[n / 2]);
	cout << ans;
	return 0;
}