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

[HAOI2008]糖果传递

题目描述

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

输入格式

小朋友个数 $n$，下面 $n$ 行 $a_i$。

输出格式

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

输入输出样例

样例输入1

4
1
2
5
4

样例输出1

4

说明 / 提示

对于 $100\%$ 的数据 $n\le 10^6$。

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;
}