洛谷P2512 [HAOI2008]糖果传递

 //不开long long见祖宗！！！
 #include<bits/stdc++.h>
 using namespace std;
 long long n,ans,sum;
 long long a[],s[];
 int main()
 {
     scanf("%lld",&n);
     for(int i=;i<=n;i++) {scanf("%lld",&a[i]);sum+=a[i];}
     sum/=n;
     for(int i=;i<=n;i++) a[i]-=sum;
     for(int i=;i<=n;i++) s[i]=s[i-]+a[i];
     nth_element(s+,s+(n+)/,s++n);//STL根据第n大元素O(n)排序
     for(int i=;i<=n;i++) ans+=s[i]>s[(n+)/]?s[i]-s[(n+)/]:s[(n+)/]-s[i];
     printf("%lld",ans);
     return ;
 }