这题就是一个强化版均分纸牌,如果知道题没有环形,那么很水。
我们现在设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);
}