BZOJ 1045 [HAOI2008] 糖果传递

1045: [HAOI2008] 糖果传递

Time Limit: 10 Sec  Memory Limit: 162 MB
Submit: 2786  Solved: 1218
[Submit][Status][Discuss]

Description

有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。

Input

小朋友个数n 下面n行 ai

Output

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

Sample Input

4
1
2
5
4

Sample Output

4

HINT

100% n<=987654321

 

Source

题解：窝也只能写这种题了。。。跟均分纸牌一样搞出一个p数组，求个中位数。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<algorithm>
#include<stack>
#include<queue>
#include<cstring>
#define PAU putchar(' ')
#define ENT putchar('
')
using namespace std;
const int maxn=1000000+10;
int A[maxn],p[maxn],n;
long long ans,tot,pos;
inline int read(){
    int x=0,sig=1;char ch=getchar();
    for(;!isdigit(ch);ch=getchar())if(ch=='-')sig=0;
    for(;isdigit(ch);ch=getchar())x=10*x+ch-'0';
    return sig?x:-x;
}
inline void write(long long  x){
    if(x==0){putchar('0');return;}if(x<0)putchar('-'),x=-x;
    int len=0;long long buf[20];while(x)buf[len++]=x%10,x/=10;
    for(int i=len-1;i>=0;i--)putchar(buf[i]+'0');return;
}
int main(){
    n=read();for(int i=0;i<n;i++)A[i]=read(),tot+=A[i];
    tot/=n;for(int i=1;i<n;i++)p[i]=p[~-i]+tot-A[i];
    sort(p,p+n);pos=p[n>>1];
    for(int i=0;i<n;i++)ans+=abs(p[i]-pos);write(ans);
    return 0;
}