2354: 分班级【二分】

题目描述
Zoro是一个有强迫症的人，他喜欢均衡。
17级的新生要开学了，起初所有班级是按照学生的来源地分的，各班人数非常不合理。于是老师要求Zoro来让各班人数均衡一下。
由于学校系统陈旧，每次调换只能是Zoro自己手动的把一个学生从一个班级拉到另外一个班级，由于Zoro有强迫症，他每次会找出班级人数最多的那个拉出一个人转到一个人数最少的班级，而且每次操作，耗费一次权限，而他的账号只有k次权限。老师最后会看所有班级人数差的最大值。
由于班级实在太多，Zoro要做完需要很长时间，你能不能帮助Zoro先计算出他工作完成后所有班级人数差的最大值报告给老师。

输入
第一行输入两个整数n和k，分别代表班级数量和Zoro账号的操作权限次数。
接下来一行n个整数 第i个数字代表第i个班级有ci个人。
(1<=n<=500000,0<=k<=1e9,0<= ci <=1e9)
输出
输出一个整数表示最后所有班级人数差的最大值。
样例输入
5 1
1 2 3 4 5
样例输出
2
题目链接

思路：
找出变化后最大值和最小值的分界点，其实就是平均值，根据这个两次二分：
第一次二分人数少的班级区间，找出满足使人数少的班级增加k个后的最小值；
第二次二分人数多的班级区间，找出满足使人数多的班级减少k个后的最大值；
需要注意的是平均值可能不是整数，那么无论k为何值最后结果都不可能是0，注意第二次二分的左端点；

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath>
typedef long long LL;
using namespace std;
LL n, k, sum = 0;
LL a[500010];

bool judge1(LL x) {
    LL ans = 0;
    for(int i = 1; i <= n; i++) {
        if(x > a[i]) ans += x - a[i];
    }
    return ans <= k;
}

bool judge2(LL x) {
    LL ans = 0;
    for(int i = 1; i <= n; i++) {
        if(x < a[i]) ans += a[i] - x;
    }
    return ans <= k;
}

int main() {
    scanf("%lld %lld", &n, &k);
    for(int i = 1; i <= n; i++) {
        scanf("%lld", &a[i]); 
        sum += a[i];
    }
    LL len = sum / n;
    LL res1, res2;
    LL l = 1, r = len, mid;
    while(r - l >= 0) {
        mid = (l + r) >> 1;
        if(judge1(mid)) {
            res1 = mid;
            l = mid + 1;
        }
        else r = mid - 1;
    }
    l = len + ((sum % n) ? 1 : 0), r = sum, mid; //注意左端点
    while(r - l >= 0) {
        mid = (l + r) >> 1;
        if(judge2(mid)) {
            res2 = mid;
            r = mid - 1;
        }
        else l = mid + 1;
    }
//  printf("%lld %lld\n", res1, res2);
    printf("%lld\n", res2 - res1);
    return 0;
}