一道神奇的题。。看到做法是排序我的心是绝望的。。
首先我们可以先求出每个小朋友应该得到的糖果数,就是平均值,然后ave-a[i]就代表要从其他小朋友那得到多少个糖果(如果是负数就是要送出糖果)然后求前缀和,往前面延伸,将值想象成数轴上的点,数轴上任意点到其他的值的点的最小距离和就在中位数。
#include<cstdio> #include<cstring> #include<algorithm> using namespace std; typedef long long LL; int a[1100000],p[1100000]; int main() { int n;LL ave=0; scanf("%d",&n); for(int i=1;i<=n;i++) { scanf("%d",&a[i]); ave+=a[i]; } ave/=n; for(int i=2;i<=n;i++)p[i]=p[i-1]+ave-a[i]; sort(p+1,p+n+1); LL ans=0; for(int i=1;i<=n;i++)ans+=abs(p[i]-p[(n+1)/2]); printf("%lld ",ans); return 0; }