mxj的新年礼物&&贪心

题目背景

在新的一年,mxj给俱乐部的孩子们准备了一堆的礼物,它生成了一个随机数列表aa,使得第i个人会得到a[i]件礼物。
但是,因为某些人得到的礼物太多,为了公平起见,mxj建议大家把礼物平均分配。

题目描述

大家围成了一个圈,每个人可以向他左边或者右边的人传递礼物,问,至少传递多少件礼物,才能让每个人得到的礼物件数相同呢?(保证礼物总数是人数n的倍数)

输入格式:

第一行输入一个正整数n,表示俱乐部成员的个数。

接下来有n行,每行一个整数a[i],表示第i个成员初始得到的礼物的件数。

输出格式:

输出一个整数,表示至少需要传递多少件礼物。

输入样例#1:

4
1
2
5
4

输出样例#1:

4

说明

1≤n≤1000000
0≤a[i]≤109

解题思路

简单的贪心问题,不多bb。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long LL;
LL a[1000005];
LL l[1000005]={0};
int main()
{
    LL n,sum=0;
    scanf("%lld",&n);
    for(int i=0; i<n; i++)
    {
        scanf("%lld",&a[i]);
        sum+=a[i];
    }
    sum/=n;

    LL ans=0;
    for(int i=0;i<n;i++)
        a[i]-=sum;
    for(int i=0;i<n;i++)
        l[i]=l[i-1]+a[i];
    sort(l,l+n);
    for(int i=0;i<n;i++)
        ans+=abs(l[n/2]-l[i]);
    printf("%lld\n",ans);
    return 0;
}

忘了当初咋做的了T^T

猜你喜欢

转载自blog.csdn.net/weixin_43105110/article/details/87947577