糖果传递 贪心,数学

有n个小朋友坐成一圈,每人有a[i]个糖果。

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

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

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

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

接下来n行,每行一个整数a[i],表示第i个小朋友初始得到的糖果的颗数。
输出格式

输出一个整数,表示最小代价。
数据范围

1≤n≤1000000

数据保证一定有解。
输入样例:

4
1
2
5
4

输出样例:

4
设每个小朋友给出的糖果为xi,那么就可以求出每个人给出的糖果和第一个小朋友的关系,要求的是所有x绝对值加起来的最小值,xi-x1会是一个常数,就把所有常数排序取中位数,求出最小的代价。

#include<bits/stdc++.h>
using namespace std;
const   int N=1000010;
int a[N];
int c[N]; int main()
{
   
    int n;
    cin>>n;
    for(int i=1;i<=n;i++)
    cin>>a[i];
    long long sum=0;
    for(int i=1;i<=n;i++)
    sum+=a[i];
    int aver=sum/n;
    c[n]=a[n]-aver;
    for(int i=n-1;i>1;i--)
    c[i]=c[i+1]+a[i]-aver;
    c[1]=0;
    sort(c+1,c+1+n);
    long long res=0;
    for(int i=1;i<=n;i++)
    res+=abs(c[i]-c[(n+1)/2]);
    cout<<res<<endl;
    return 0;
}
全部评论

相关推荐

07-07 11:33
江南大学 Java
已经在暑假实习了&nbsp;,没有明确说有hc,纠结实习到八月份会不会有点影响秋招毕竟感觉今年好多提前批
程序员小白条:92的话准备提前批,其他没必要,没面试机会的,而且你要准备充分,尤其八股和算法题
点赞 评论 收藏
分享
码农索隆:想看offer细节
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务