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 = 500000 + 10;
const int INF = 0x3f3f3f3f;
typedef __int64 LL;

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

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

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

int main() {
    while (scanf("%d%d", &n, &k) == 2 && n) {
        init();
        LL sum = 0;
        int mi, ma;
        for (int i = 0; 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 = 0;
        if (mi >= ma) {
            if (sum%n) ans = 1;
        }
        else {
            ans = ma - mi;
        }
        printf("%d\n", ans);
    }
    return 0;
}

 

转载于:https://www.cnblogs.com/fenice/p/5549664.html