糖果传递（贪心、数学推导）

最新推荐文章于 2024-05-11 18:41:21 发布

念兮兮

最新推荐文章于 2024-05-11 18:41:21 发布

阅读量68

点赞数

分类专栏：《信息学奥赛一本通-提高篇》在线题库文章标签： c++ 算法开发语言

本文链接：https://blog.csdn.net/qq_52081267/article/details/129689735

版权

《信息学奥赛一本通-提高篇》在线题库专栏收录该内容

40 篇文章 0 订阅

订阅专栏

原题链接

解题思路：先求出每位同学与平均数的差值b[i]，然后对每位同学的差值进行前缀和sb[i]。其次sort一下前缀和sb[i]，然后在累加一下abs(sb[i] - sb[(n + 1) / 2])即为答案

C++代码

#include <bits/stdc++.h>
#define int long long
#define LL long long
using namespace std;
const int N = 1e6 + 10;
//a为初始时每位同学拥有的糖果数，b为初始值与平均值的差值
//sb为差值的前缀和
int a[N], b[N], sb[N]; 
int n;
signed main()
{
    cin >> n;
    int sum = 0;
    for (int i = 1; i <= n; i ++ ) 
    {
        cin >> a[i];;
        sum += a[i];
    }
    for (int i = 1; i <= n; i ++ ) 
    {
        b[i] = a[i] - sum / n;
        sb[i] = sb[i - 1] + b[i];
    }
    sort(sb + 1, sb + n + 1);
    int res = 0;
    for(int i = 1; i <= n; i ++ )
    {
        res += abs(sb[i] - sb[(n + 1) / 2]);
    }
    cout << res << endl;
    return 0;
}

参考博客