codeforces 671B - Robin Hood

题目来源：http://codeforces.com/problemset/problem/671/B

题目大意：给定n个数，每次可执行一次操作，将最大的数-1，最小的数+1。最多可执行k次。问最小的极差为多少。

思路：

  对n个数求取平均数，由于操作一定是从大于平均数的数中抽取，将数转至小于平均数的数中。因此可以对最大值和最小值进行二分，验证在k次操作之内能否将所有数修改为全部大于（小于）某数。

注意：

  当平均数不是整数时，特别注意在对大于平均数的数进行二分时，一定要取(sum/n)上取整的数。

代码：

#include <bits/stdc++.h>
using namespace std;
#define ll long long
ll n,k;
const int maxn=5e5+10;
ll a[maxn];
ll ave;
bool can_up(ll x)
{
    ll tot=0;
    for(int i=n;i>=1;--i)
    {
        if(a[i]>x)
            tot+=a[i]-x;
        else break;
        if(tot>k)return 0;
    }
    if(tot>k)return 0;
    return 1;
}
bool can_down(ll x)
{
    ll tot=0;
    for(int i=1;i<=n;++i)
    {
        if(a[i]<x)
            tot+=x-a[i];
        else break;
        if(tot>k)return 0;
    }
    if(tot>k)return 0;
    return 1;
}
int main()
{
    scanf("%lld%lld",&n,&k);
    ll sum=0;
    for(int i=1;i<=n;++i)
    {
        scanf("%lld",&a[i]);
        sum+=a[i];
    }
    sort(a+1,a+1+n);
    ll ansl,ansr;
    ll l,r;
    if(sum%n==0)ave=sum/n;
    else ave=sum/n+1;
    l=ave,r=1e9+10;
    while(l<r)
    {
        ll mid=(l+r)/2;
        if(can_up(mid))r=mid;
        else l=mid+1;
    }
    ansr=l;
    ave=sum/n;
    l=0,r=ave;
    while(l<r)
    {
        ll mid=(l+r+1)/2;
        if(can_down(mid))l=mid;
        else r=mid-1;
    }
    ansl=l;
    printf("%lld\n",ansr-ansl);
    return 0;
}