609C Load Balancing

题意：

”平均数“的意思是：最大数和最小数之间的差值为0或1；

先求“平均”数组，再相减。

#include<iostream>
#include<cstdlib>
#include<cstdio>
#include<algorithm>
#include<math.h>
#include<memory.h>
#define clc(a,b) memset(a,b,sizeof(a))
#define ll long long int
using namespace std;

int n;
int a[100010];

int main()
{
    while(~scanf("%d",&n))
    {
        clc(a,0);
        ll sum=0;
        ll ans=0;
        for(int i=0; i<n; i++)
        {
           scanf("%d",&a[i]);
           sum+=a[i];
        }
        sort(a,a+n);
        if(sum%n==0)
        {
            int ave=sum/n;
            for(int i=0;i<n;i++)
                  ans+=fabs(ave-a[i]);
             ans=ans/2;
        }
        else
        {
            ll res=sum%n;
            ll ave=(sum-res)/n;
            //int vv=ave;
            int b[100010];
            //memset(b,ave,sizeof(b));
            for(int i=0;i<n;i++)
                b[i]=ave;
//            cout<<ave<<endl;
//            cout<<b[1]<<endl;
            for(int i=n-1;i>n-1-res;i--)
            {
                b[i]+=1;
            }
//            cout<<b[1]<<endl;
//            for(int i=0;i<n;i++)
//            {
//                cout<<b[i]<<endl;
//            }
            for(int i=0;i<n;i++)
            {
                   ans+=fabs(b[i]-a[i]);
            }
            ans/=2;
        }
        cout<<ans<<endl;
    }
    return 0;
}