二分答案 - 题目详情 - HydroOJ

问题描述

给定一个由n个数构成的序列a，你可以进行k次操作，每次操作可以选择一个数字，将其+1，问k次操作以后，希望序列里面的最小值最大。问这个值是多少。

输入格式

第一行输入两个正整数n,k。(1≤n≤105,1≤k≤1013)
第二行输入n个正整数，表示序列a。(1≤ai​≤106)

输出格式

输出最大的最小值

样例输入

5 10
1 4 2 6 8
Copy

样例输出

5
Copy

思路:

枚举最小值的最大值，然后判断是否可以成功，也就是使用操作次数是否在k范围内。

代码：
 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll; 
const ll N = 1e5 + 10;
ll n, k;
ll a[N];

bool check(ll mid, ll k) 
{
    ll sum = 0;
    for (ll i = 1; i <= n; i++) 
	{
        if (a[i] < mid) 
		{
            sum += mid - a[i];
        }
    }
    return sum <= k;
}

int main() {
    cin >> n >> k;
    ll min_val = LLONG_MAX;
    for (ll i = 1; i <= n; i++) 
	{
        cin >> a[i];
        min_val = min(min_val, a[i]);
    }
    sort(a + 1, a + n + 1);
    ll l = min_val, r = 1e6 + k;
    while (l + 1 != r) 
	{
        ll mid = (l + r) / 2;
        if (check(mid, k)) 
		{
            l = mid;
        } 
		else 
		{
            r = mid;
        }
    }
    cout << l;
    return 0;
}