Codeforces Round #352 (Div. 2) D. Robin Hood

题目链接：

http://codeforces.com/contest/672/problem/D

题意：

给你一个数组，每次操作，最大数减一，最小数加一，如果最大数减一之后比最小数加一之后要小，则取消操作，现在给你操作的次数，问操作之后最大数减最小数的最小值。

题解：

问题要求得是min(k次操作之后的最大数-k次操作之后的最小数），而这两个数可以独立求出来，我们先用二分求k次操作之后的最小数的最大取值，然后，再用二分求k次操作之后的最大数的最小可能取值。

代码：

#include<algorithm>
#include<iostream>
#include<cstring>
#include<vector>
#include<queue>
using namespace std;

const int maxn =  + ;
const int INF = 0x3f3f3f3f;
typedef __int64 LL;

LL arr[maxn];
int n, k,Ma,Mi;

bool ok1(int x) {
    LL cnt = ;
    for (int i = ; i < n; i++) {
        cnt += max((LL), x - arr[i]);
    }
    if (cnt <= k) return true;
    return false;
}
int bs1() {
    int l = Mi,r = Ma+;
    while (l +  < r) {
        int mid = l + (r - l) / ;
        if (ok1(mid)) l = mid;
        else r = mid;
    }
    return l;
}
bool ok2(int x) {
    LL cnt = ;
    for (int i = ; i < n; i++) {
        cnt += max((LL), arr[i]-x);
    }
    if (cnt <= k) return true;
    return false;
}
int bs2() {
    int l = Mi-, r = Ma;
    while (l +  < r) {
        int mid = l + (r - l) / ;
        if (ok2(mid)) r = mid;
        else l = mid;
    }
    return r;
}

void init() {
    Mi = INF; Ma = -;
}

int main() {
    while (scanf("%d%d", &n, &k) ==  && n) {
        init();
        LL sum = ;
        int mi, ma;
        for (int i = ; i < n; i++) {
            scanf("%I64d", arr + i);
            sum += arr[i];
            Ma = max((LL)Ma, arr[i]), Mi = min((LL)Mi, arr[i]);
        }
        mi = bs1();
        ma = bs2();
        int ans = ;
        if (mi >= ma) {
            if (sum%n) ans = ;
        }
        else {
            ans = ma - mi;
        }
        printf("%d\n", ans);
    }
    return ;
}