AcWing 122. 糖果传递
宋标 Lv5
  2022-08-18 02:31:35 2022-08-18 02:31    

题目

有 个小朋友坐成一圈，每人有 个糖果。

每人只能给左右两人传递糖果。

每人每次传递一个糖果代价为 。

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

输入格式

第一行输入一个正整数 ，表示小朋友的个数。

接下来 行，每行一个整数 ，表示第 个小朋友初始得到的糖果的颗数。

输出格式

输出一个整数，表示最小代价。

数据范围

,
,
数据保证一定有解。

输入样例：

4
1
2
5
4

输出样例：

4

题解

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
#include <iostream>
#include <algorithm>

using namespace std;

const int N = 1000010;

int n;
int a[N];

int main()
{
    long long res = 0, sum = 0, avg = 0;

    scanf("%d", &n);
    for (int i = 1; i <= n; ++ i) scanf("%d", &a[i]), sum += a[i];

    avg = sum / n;

    for (int i = n; i > 1; -- i)
        a[i] = a[i] - avg + a[i + 1];

    // 为什么a[1]等于0，没搞懂
    a[1] = 0;

    // 递归快排会超时
    sort(a + 1, a + n + 1);

    for (int i = 1; i <= n; ++ i) res += abs(a[i] - a[(n +  1) / 2]);

    cout << res << endl;

    return 0;
}
 评论