[HAOI2008] 糖果传递

这题就是一个强化版均分纸牌，如果知道题没有环形，那么很水。

我们现在设A为平均数，设f[i]表示第i个要给i-1个多少个糖果，f[1]代表第1个要给第n个多少个，有f[i+1]=f[i]+A-a[i] ,有f[i+1]-f[i]=A-a[i]

我们设w[i]=A-a[i-1] 有f[i+1]-f[i]=w[i+1] 可得有:

f[n]-f[n-1]=w[n]; .........(n-1)

f[n-1]-f[n-2]=w[n-1]; .......(n-2)

f[n-2]-f[n-3]=w[n-2]; ........(n-3)

... ... ...

f[2]-f[1]=w[2];  .........(1)

我们把前i个等式相加，可以得到f[i]-f[1]=s[i]; s[i]=sigma(w[j]|j<=i,w[1]=0)

f[i]=s[i]+f[1];

而ans=sigma(f[i]);

说以我们要找到一个f[1]使sigma(s[i]+f[1])最小，明显f[1]就等于s[i]中位数的相反数

附代码

#include<cstdio>
#include<iostream>
#include<algorithm>
#define judgeonline
const int maxn=1000010;
using namespace std;
  
#ifdef judgeonline
const int maxchar=10000000;
char buf[maxchar],*wbuf=buf;
inline int getint(){
    int tmp=0;
    while (*wbuf<'0' || *wbuf>'9') ++wbuf;
    for (;*wbuf>='0' && *wbuf<='9';++wbuf) tmp=tmp*10+*wbuf-'0';
    return tmp;
}
#else
inline int getint(){
    char ch=getchar(); int tmp=0;
    while (ch<'0' || ch>'9') ch=getchar();
    for (;ch<='9' && ch>='0';ch=getchar()) tmp=tmp*10+ch-'0';
    return tmp;
}
#endif
  
int find(int *l,int *r,int rank){
    int *i=l,*j=r;
    int mid=*(i+(j-i)/2);
    while (i<=j){
        while (*i<mid) ++i;
        while (*j>mid) --j;
        if (i<=j){
            swap(*i,*j);
            ++i,--j;
        }
    }
    if (j-l+1>=rank) return find(l,j,rank);
    if (rank>i-l) return find(i,r,rank-(i-l));
    return *(l+rank-1);
}
  
int n;
int a[maxn],p[maxn];
int main(){
#ifdef judgeonline 
    fread(wbuf,1,maxchar,stdin);
#endif
    n=getint();
    long long sum=0;
    for (int i=1;i<=n;++i){
        a[i]=getint();
        sum+=a[i]; 
    }
    int k=sum/n;
    p[1]=k-a[1];
    for (int i=2;i<=n;++i)
        p[i]=p[i-1]+k-a[i];
//  sort(p,p+n);
    int tmp=find(p+1,p+n,(n+1)/2);
    long long ans=0;
    for (int i=1;i<=n;++i)
        ans+=abs(p[i]-tmp);
    printf("%I64d\n",ans);
}