Codeforces 609C Load Balancing

最新推荐文章于 2025-02-01 20:26:29 发布

原创最新推荐文章于 2025-02-01 20:26:29 发布 · 438 阅读

0 ·

CC 4.0 BY-SA版权

本文详细解析 CodeForces 平台上的 609C 题目，介绍一种通过计算平均值来确定最优解的算法策略。该策略包括将序列中小于平均值的所有元素提升到平均值，或者将大于平均值加一的所有元素降低到平均值加一，从而确保序列的最大最小值之差不超过1。

题目链接：http://codeforces.com/problemset/problem/609/C

题意：将一个大小为n序列重新分配，将ai-1同时aj+1算一次操作，问要多少次操作后序列中min和max的差不大于1

思路：一开始想了个很蠢的方法，各种wa，其实就只有2种情况，设平均值为x，要么把所有小于x的数变成x，要么把大于x+1的数变成x+1，分别计算这2中情况的操作数，选最大的一个

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>
#define LL long long
using namespace std;

LL s[100030];

int main()
{
    int n;
    while (scanf("%d",&n)!=EOF)
    {
        LL sum=0;
        for (int i=0;i<n;i++)
        {
            scanf("%I64d",&s[i]);
            sum+=s[i];
        }
        sum=sum/n;
        sort(s,s+n);
        LL need=0,res=0;
        for (int i=0;i<n;i++)
        {
            if (s[i]<sum)
            {
                res+=sum-s[i];
                need+=sum-s[i];
            }
            else if (s[i]>sum+1)
            {
                res+=s[i]-1-sum;
                need-=s[i]-1-sum;
            }
        }
        printf("%I64d\n",(res+need)/2);
    }
}