codeforces 672 D

题目链接：http://codeforces.com/problemset/problem/672/D

题目大意：进行k次操作，每次将最大值集合中最大值-1，最小值+1,问你K次操作之后，最大值和最小值的差值是多少

思路分析：二分寻找最大值和最小值，check函数很好写，但是关键的一点需要注意，就是二分的上下限，当k非常大

时,最后的结果就已经稳定在平均数位置，如果二分上下限没有选对，就会出现最小值>最大值的情况orz

代码：

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=500000+100;
typedef long long ll;
int a[maxn];
int n,k;
int checkmi(int x)
{
    ll sum=0;
    for(int i=0;i<n;i++)
    {
        if(a[i]>x) break;
        sum+=x-a[i];
    }
    if(sum<=k) return 1;
    else return 0;
}
bool checkma(int x)
{
    ll sum=0;
    for(int i=n-1;i>=0;i--)
    {
        if(a[i]<x) break;
        sum+=a[i]-x;
    }
    if(sum<=k) return 1;
    else return 0;
}
int main()
{
    while(scanf("%d%d",&n,&k)!=EOF)
    {
        ll sum=0;
        for(int i=0;i<n;i++)
            {scanf("%d",&a[i]);
             sum+=a[i];}
        sort(a,a+n);
        int al,ar;
        if(sum%n==0) {al=sum/n,ar=sum/n;}
        else {al=sum/n,ar=sum/n+1;}
        int l=a[0],r=al;
        int ansl=a[0],ansr=a[0];
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(checkmi(mid))
            {
                 ansl=mid;
                l=mid+1;
            }
            else r=mid-1;
        }
        l=ar,r=a[n-1];
        while(l<=r)
        {
            int mid=(l+r)>>1;
            if(checkma(mid))
            {
                ansr=mid;
                r=mid-1;
            }
            else l=mid+1;
        }
        printf("%d\n",ansr-ansl);
    }
    return 0;
}